tensorflow1.X读取checkpoint(加载训练好的模型及参数)的两种方式

已于 2022-04-01 16:29:24 修改
阅读量2.8k 收藏 5
点赞数 1
分类专栏： tensorflow python 文章标签： tensorflow python
于 2022-03-25 10:57:10 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_43220532/article/details/123729437
版权

在模型训练的保存路径下面会保存有多个模型（如下图），下面介绍两种读取方式
在这里插入图片描述
1，获取最新的保存的一个模型，比如上图的model-5200

# model_path是checkpoints的路径，下面方法会自动寻找最新保存的一个模型
ckpt_file = tf.train.latest_checkpoint(model_path)
saver.restore(sess,ckpt_file)

2，获取某个迭代次数的模型，比如要指定获取上图的model-5199

ckpt_file=os.path.join(model_path,'model-5199')
saver.restore(sess,ckpt_file)

Done!!!

CrystalheartLi
关注
  • 1
    点赞
  • 踩
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
专栏目录
TensorFlow实现模型断点训练,checkpoint模型载入方式
09-16
主要介绍了TensorFlow实现模型断点训练,checkpoint模型载入方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
tensorflow实现训练变量checkpoint的保存与读取
12-20
1.保存变量 先创建（在tf.Session()之前）saver saver = tf.train.Saver(tf.global_variables(),max_to_keep=1) #max_to_keep这个保证只保存最后一次training的训练数据 然后在训练的循环里面 checkpoint_path = os.path.join(Path, ‘model.ckpt’) saver.save(session, checkpoint_path, global_step=step) #这里的step是循环训练的次数，也就是第几次迭代 以下保存的变量文件 2.变量读取 1.若要直接恢
tensorflow测试时读取任意指定的check point的网络参数
云台
08-23 2445
tensorflow在训练时会保存三个文件， model.ckpt-xxx.data-00000-of-00001 model.ckpt-xxx.index model.ckpt-xxx.meta 其中第一个储存网络参数值，第二个储存每一层的名字，第三个储存图结构 随着训练的过程，每隔一段时间都会保存一组以上三个文件，而在训练之前我们并不知道什么时候可以达到最佳的拟合，训练时间过短会导致欠...
tensorflow保存加载模型查看训练参数
James_Ying的专栏
04-18 1万+
TensorFlow的checkpoint机制使得其能够同时支持Online Learning和Continuous Learning，首先，通过tf.train.Saver()将训练好的或者训练过程中的模型保存成checkpoint:_, loss_value, step = sess.run([train_op, loss, global_step]) saver.save(sess,"./c
keras.callbacks.ModelCheckpoint()
baixue0729的博客
07-17 5251
每个纪元后保存模型。 filepath可以包含命名格式化选项，它将填充日志中的纪元和键的值（在on_epoch_end中传递）。 例如：如果filepath是权重。{epoch：02d} - {val_loss：.2f} .hdf5，那么模型检查点将与文件名中的纪元号和验证丢失一起保存。 def __init__(self, filepath: Any, ...
tensorflow加载训练好的模型及参数(读取checkpoint)
day day up
02-18 4632
checkpoint 保存路径 model_path下存有包含多个迭代次数的模型 1.获取最新保存的模型 即上图中的model-9400 import tensorflow as tf graph=tf.get_default_graph() # 获取当前图 sess=tf.Session() sess.run(tf.global_variables_initializer()) c...
tensorflow的ckpt文件转成pb文件
qq_30066451的博客
07-25 847
最近在做深度学习的工程应用部署，需要使用C++调用tensorflow的API使用训练好的模型进行预测，所以需要把checkpoint的文件转成pb文件做离线调用。 1. ckpt文件 2. 获得输入输出节点 转pb文件时需要知道确切的输出节点，调用时需要知道输入节点。由ckpt文件获取输入输出节点 2.1 使用python获得所有节点 import tensorflow as...
tensorflow保存和读取模型(通过图.meta)
赵凯月的博客
08-15 4317
1、加载模型 当我们基于checkpoint文件(ckpt)加载参数时，实际上我们使用Saver.restore取代了initializer的初始化 checkpoint文件会记录保存信息，通过它可以定位最新保存的模型： checkpoint文件会记录保存信息，通过它可以定位最新保存的模型： ckpt = tf.train.get_checkpoint_state('./model/') print(ckpt) 结果如下： tf.train.get_checkpoint_state('./model/
TensorFlow2.0——模型保存、读取与可训练参数提取
weixin_43499703的博客
04-10 336
https://www.cnblogs.com/dwithy/p/14015501.html?ivk_sa=1024320u
Tensorflow加载预训练模型和保存模型（ckpt文件）以及迁移学习finetuning
热门推荐
loveliuzz的博客
08-14 10万+
转载自：https://blog.csdn.net/huachao1001/article/details/78501928 使用tensorflow过程中，训练结束后我们需要用到模型文件。有时候，我们可能也需要用到别人训练好的模型，并在这个基础上再次训练。这时候我们需要掌握如何操作这些模型数据。看完本文，相信你一定会有收获！ 1 Tensorflow模型文件 我们在checkpoint_d...
基于tensorflow1.X版本的中文语言模型（NLP项目）
06-21
### 语言模型 #### 数据预处理 中文语言模型基本都是基于字的模型，因此不需要做太多的操作 #### 文件结构介绍 * config文件：配置各种模型的配置参数 * data：存放训练集和测试集 * data_helpers：提供数据处理的...
tensorflow模型的save与restore,及checkpoint中读取变量方式
09-16
主要介绍了tensorflow模型的save与restore,及checkpoint中读取变量方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
ValueError: Connection error, and we cannot find the requested files in the cached path. Please...
weixin_43220532的博客
04-12 1万+
Python使用transformers包的以下l两个函数加载bert时候， BertTokenizer.from_pretrained(bert_path) BertModel.from_pretrained(bert_path) 报错： ValueError: Connection error, and we cannot find the requested files in the cached path. Please try again or make sure your Internet c
解决: ...HTTPError:404 Client Error: Not Found for url: https://pypi.tuna.tsinghua.edu.cn/simple/pip3/
weixin_43220532的博客
05-07 1万+
更新pip3 报错：requests.exceptions.HTTPError: 404 Client Error: Not Found for url: https://pypi.tuna.tsinghua.edu.cn/simple/pip3/ 具体报错如下图： 因为我使用的更新命令为： $ pip3 install -U pip3 解决办法为，将命令改为： $ pip3 install -U pip Done！！！ ...
pip3 install XXX报错“python setup.py egg_info Check the logs for full command output.“
weixin_43220532的博客
01-25 3779
具体报错如下图： 出错原因：pip3 与 setuptools版本比较低 解决办法： $ python3 -m pip install --upgrade setuptools $ python3 -m pip install --upgrade pip Done!
tensorflow.python.framework.errors_impl.DataLossError:Unable to open table file xxx perhaps ........
weixin_43220532的博客
05-10 2434
tensorflow加载预训练模型报错： tensorflow.python.framework.errors_impl.DataLossError: Unable to open table file /home/chinese_L-12_H-768_A-12/bert_model.ckpt.data-00000-of-00001: Data loss: not an sstable (bad magic number): perhaps your file is in a different file
ModuleNotFoundError: No module named 'tensorflow.python.checkpoint'
最新发布
08-02
根据你提供的信息，报错信息"ModuleNotFoundError: No module named 'tensorflow.python.checkpoint'"是由于缺少tensorflow.python.checkpoint模块导致的。这个问题可能是由于tensorflow-estimator版本不对或者...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

CrystalheartLi

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值