tensorflow中加載模型兩種方法

? ? 在使用tensorflow過程中,把模型保存下來稍后使用,是常見的需求。初步研究,各種資料混雜,讀者容易不知所云,這里介紹兩種方法。

本文方法在 TensorFlow 1.4 版本基礎(chǔ)上實(shí)驗(yàn)。




首先定義一個(gè)簡單的模型,分別采用兩種方法還原模型。

train.py

這個(gè)簡單的模型是 y = w + x. 其中 x 是輸入, y 是輸出, w采用了隨機(jī)一個(gè)權(quán)重。

輸出結(jié)果:

w [[-0.07159261 -0.33205539]]

y [[ 5.92840719? 6.66794443]]

現(xiàn)在可以看到,文件夾下多了4個(gè)文件:



1. 還原變量的角度

把W還原出來,然后可以自己定義 y = w + x

輸出結(jié)果?[[-0.07159261 -0.33205539]],可見,這就是 w。

然后就可以利用 w ,重構(gòu)模型。

這個(gè)方法需要重新定義原模型中的 變量,加載模型后,重新定義的變量被保存的值賦值。

2. 還原圖的角度

如果不想重新定義變量,可以把圖還原,如下:

輸出結(jié)果:

w [[-0.07159261 -0.33205539]]

[[ 4.92840719? 5.66794443]]

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • 本系列文章面向深度學(xué)習(xí)研發(fā)者,希望通過Image Caption Generation,一個(gè)有意思的具體任務(wù),深入...
    imGeek閱讀 2,001評論 0 8
  • 一、實(shí)驗(yàn)?zāi)康?學(xué)習(xí)使用 weka 中的常用分類器,完成數(shù)據(jù)分類任務(wù)。 二、實(shí)驗(yàn)內(nèi)容 了解 weka 中 explo...
    yigoh閱讀 8,845評論 5 4
  • 今天是星期二,也是軍訓(xùn)的第十四天。今天,我們排練了明天的結(jié)營儀式,我感覺我們排練的還是不錯的。排練的內(nèi)容有手語操,...
    WMY王明陽閱讀 156評論 0 0
  • (一) 哲學(xué)就如長途中午夜的旅館,那一盞寂靜的燈。如激動地靈魂捧起一把故鄉(xiāng)的土壤,那熱忱的淚水潸潸落地。似乎是一場...
    博古林V雙木居士閱讀 936評論 1 6
  • 被描述的對象 行尸走肉 生活失去精神 夜的帷幕落下 才見光芒 當(dāng)饑腸轆轆 路邊的老鼠在覓食 我曾不止一次好奇 為什...
    譚劍飛閱讀 245評論 0 1

友情鏈接更多精彩內(nèi)容