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

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

恢復 Tensorflow 的中間檢查點文件

恢復 Tensorflow 的中間檢查點文件

開心每一天1111 2021-07-30 14:16:17
Tensorflow 版本 =1.8.0我正在嘗試使用 Tensorflow 中的一個中間檢查點文件來恢復我的模型。默認情況下,Tensorflow 將采用最后保存的檢查點文件。例如,該文件夾包含以下文件:checkpoint model-56000.index model-56000.data-00000-of-00001 model-56000.meta model-57000.index model-57000.data-00000-of-00001 model-57000.meta默認情況下,Tensorflow 加載最后的 57K 檢查點,但出于某些原因,我想加載 56K 檢查點的權重。以下是我用于恢復模型的代碼:def load_G(self, checkpoint_dir):    print(" [*] Reading checkpoints of G...")    ckpt = tf.train.get_checkpoint_state(checkpoint_dir)    if ckpt and ckpt.model_checkpoint_path:        ckpt_name = os.path.basename(ckpt.model_checkpoint_path)        self.saver_gen.restore(self.sess, os.path.join(checkpoint_dir, ckpt_name))        return True    else:        return False從 Tensorflow 的頁面,我讀到 tf.train.get_checkpoint_state(),我可以指定 tf.train.get_checkpoint_state(checkpoint_dir, latest_filename=None)。但我想不通,我應該為 latest_filename 寫什么。我嘗試編寫 latest_filename = model-56000 但這并沒有加載模型。我也嘗試編寫 latest_filename = model-56000.meta。那也沒有用。那么,在 Tensorflow 中加載一些中間檢查點文件的正確方法是什么?
查看完整描述

2 回答

?
千巷貓影

TA貢獻1829條經驗 獲得超7個贊

好的,所以黑客是修改檢查點protobuf 文件并將該文件的第一行從:model_checkpoint_path: "model-57000"更改為model_checkpoint_path: "model-56000" ad 現在它加載了 56K 檢查點。尋找一些更好的方法來做到這一點。


查看完整回答
反對 回復 2021-08-03
?
陪伴而非守候

TA貢獻1757條經驗 獲得超8個贊

ckpt 文件名將是 model-56000.ckpt

model-56000.meta 指向ckpt的元信息

model-56000 是 ckpt、數據文件或元文件的文件名


查看完整回答
反對 回復 2021-08-03
  • 2 回答
  • 0 關注
  • 275 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

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

幫助反饋 APP下載

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

公眾號

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