Tensorflow2.0 入门与实战学习笔记(补充)-模型保存与恢复

1 简介

此方法保存以下所有内容:
1.权重值

2.模型配置(架构)

3.优化器配置
 

1.1 model.save()

仅保存架构

恢复模型(重建):

还差保存优化器的状态

1.2 保存检查点

2 重新加载模型

在自定义训练中保存检查点

checkpoint保存变化的地方

保存地方:

猜你喜欢

转载自blog.csdn.net/qq_37457202/article/details/107977144