python - 在特定迭代或检查点将模型加载/恢复到 tensorflow

标签 python tensorflow restore

我有一个模型,我每迭代 10 次就保存一次。所以,我在保存的目录中有以下文件。

checkpoint  model-50.data-00000-of-00001  model-50.index  model-50.meta
model-60.data-00000-of-00001  model-60.index  model-60.meta

以此类推,直到 100。我只需要加载 model-50。因为我有 70 次迭代后的 NaN 值。默认情况下,当我恢复时,保护程序将寻找最终检查点。那么,我该如何专门加载 model-50.请帮忙,否则,我必须从头开始运行模型增益,这很耗时。

最佳答案

由于您正在使用 tf.train.Saver 的函数 restore(),您可以使用 last_checkpoints函数来获取所有可用检查点的列表。您将在此列表中同时看到 model-50model-60

选择正确的模型,然后像这样将其直接传递给 restore()

saver.restore(sess, ckpt_path)

关于python - 在特定迭代或检查点将模型加载/恢复到 tensorflow,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41455575/

相关文章:

javascript - AJAX 数据被发送到错误的 Django View

python - 为什么 Tensorflow Reshape tf.reshape() 会破坏梯度流?

sql - 还原 Azure DTU 中的 SQL 数据库后,保持 100% 全时状态

sql-server - 如何查询SQL Server数据库恢复需要多长时间?

php - 如何从 Time Machine 或备份恢复 Homebrew 配置

python - 如何将每一列数据与其他每一列数据进行比较

python - 扩大 Pandas 数据框,类似于 Pivot 或 Stack/Unstack

python - E : Unable to find a source package for psycopg2 or use pip with virtualenv

c++ - 构建TensorFlow r1.14 C++文件时缺少文件 “tensorflow/core/framework/types.pb.h”

machine-learning - 实现超分辨率 CNN 时遇到问题