断点续训
Posted cropmodel
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了断点续训相关的知识,希望对你有一定的参考价值。
作用:存取模型
方法:
读取模型:
1 checkpoint_save_path = ‘./checkpoint/Baseline.ckpt‘ 2 if os.path.exists(checkpoint_save_path+‘.index‘): 3 print(‘load exists model‘) 4 model.load_weights(checkpoint_save_path)
保存模型:
以上是关于断点续训的主要内容,如果未能解决你的问题,请参考以下文章
第四讲 网络八股拓展--用mnist数据集实现断点续训, 绘制准确图像和损失图像