首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >tf.train.latest_checkpoint()是否从tf.train.import_meta_graph()继承模型名称

tf.train.latest_checkpoint()是否从tf.train.import_meta_graph()继承模型名称
EN

Stack Overflow用户
提问于 2017-08-05 00:50:28
回答 1查看 405关注 0票数 0

在这种情况下,我有以下代码:

代码语言:javascript
复制
new_saver = tf.train.import_meta_graph('../logs/model-2.meta')
new_saver.restore(restored_sess,tf.train.latest_checkpoint('../logs/')) 

在我的../logs/文件中有两个模型: model-1和model-2

假设在import_meta_graph中指定了最新的检查点,TF是否会自动从model-2获取最新的检查点。

将不同的模型保存在不同的文件夹中会更好吗?

EN

回答 1

Stack Overflow用户

发布于 2017-08-05 01:38:43

您可以使用saver设置型号名称,以便对不同型号的saver.save(sess, 'model_name', global_step=epoch_number)使用单独的名称。对于一个文件夹,只保存一个checkpoint文件。这将保留tf.train.latest_checkpoint使用的最新型号的相关信息。

因此,对于不同的模型,最好使用不同的文件夹。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/45511932

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档