模型的保存与还原
有没有想过一个需要训练一个礼拜的模型
到最后一天突然断电了
那怎么办?
思路可能是多少步保存一下模型参数,
即不能训练结束之后才保存

模型的保存与还原
一般目录可能会出错的有
最后要以/结尾
比如上面如果是./ckpt_dir就会报错

还有/等价于\
还有.表示当前目录

模型的保存与还原
模型的保存与还原
os.path.join函数
format函数
模型的保存与还原
模型的保存与还原
程序中是每个5轮保存一次,保存的时候设置了序号,这样方便查阅,一次保存是保存三个文件
程序中是每个5轮保存一次,保存的时候设置了序号,这样方便查阅,一次保存是保存三个文件
程序中是每个5轮保存一次,保存的时候设置了序号,这样方便查阅,一次保存是保存三个文件

checkpoint只有一份

那这样的话会不会保存很多文件,
答案是不会
因为模型保存机制缺省(默认)的是保存最近5份

模型的保存与还原

模型的保存与还原
模型的保存与还原
模型的保存与还原
模型的保存与还原
模型的保存与还原

听说还可以在导入模型后进行断点续训

听说还可以在导入模型后进行断点续训

听说还可以在导入模型后进行断点续训

这个要到后面才会用到

相关文章: