tensorflow模型的保存与恢复

模型保存后产生四个文件,分别是:

|--models
| |--checkpoint
| |--.meta
| |--.data
| |--.index

.meta保存的是图的结构

checkpoint文件是个文本文件,里面记录了保存的最新的checkpoint文件以及其它checkpoint文件列表。

.data和.index保存的是变量值。

 

tensorflow常用的模型保存方法:

best_str = ''
if best_loss is None or valid_loss < best_loss:
  best_loss = valid_loss
  best_str = ', best!'
  early_stop_i = 0
  saver.save(sess, '%s/model_best.tf' % FLAGS.logdir)
else:
  early_stop_i += 1

代码比较简单,就是取每个轮次训练后损失最小的那一次保存模型,并进行后续加载

 

posted on   tuzhuo  阅读(297)  评论(0编辑  收藏  举报

编辑推荐:
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
阅读排行:
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

点击右上角即可分享
微信分享提示