亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

tensorflow怎么調用ckpt繼續訓練 ?

tensorflow怎么調用ckpt繼續訓練 ?

三國紛爭 2018-10-14 17:13:44
tensorflow怎么調用ckpt繼續訓練 
查看完整描述

1 回答

?
智慧大石

TA貢獻1946條經驗 獲得超3個贊

訓練完一個模型后,為了以后重復使用,通常我們需要對模型的結果進行保存。如果用Tensorflow去實現神經網絡,所要保存的就是神經網絡中的各項權重值。建議可以使用Saver類保存和加載模型的結果。

1、使用tf.train.Saver.save()方法保存模型

  • sess: 用于保存變量操作的會話。

  • save_path: String類型,用于指定訓練結果的保存路徑。

  • global_step: 如果提供的話,這個數字會添加到save_path后面,用于構建checkpoint文件。這個參數有助于我們區分不同訓練階段的結果。

  • 2、使用tf.train.Saver.restore方法價值模型

    • sess: 用于加載變量操作的會話。

    • save_path: 同保存模型是用到的的save_path參數。

    • 下面通過一個代碼演示這兩個函數的使用方法

    假設保存變量的時候是

checkpoint_filepath='models/train.ckpt'saver.save(session,checkpoint_filepath)則從文件讀變量取值繼續訓練是saver.restore(session,checkpoint_filepath)



查看完整回答
反對 回復 2018-11-11
  • 1 回答
  • 0 關注
  • 2545 瀏覽

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號