函數功能：找出訓練時保存的模型

ckpt.model_checkpoint_path可以找出所有模型中最新的模型

ckpt = tf.train.get_checkpoint_state('/mnist/summary/train') if ckpt and ckpt.model_checkpoint_path:print(ckpt.model_checkpoint_path)

輸出：

/mnist/summary/train/model.ckpt-1000（1000是訓練步數）

ckpt.all_model_checkpoint_paths可以找出所有模型

ckpt = tf.train.get_checkpoint_state('mnist/summary/train') if ckpt and ckpt.model_checkpoint_path:print(ckpt.all_model_checkpoint_paths)

輸出：

['mnist/summary/train/model.ckpt-500', 'mnist/summary/train/model.ckpt-1000']

總結

以上是生活随笔為你收集整理的tf.train.get_checkpoint_state的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。