有没有想过一个需要训练一个礼拜的模型
到最后一天突然断电了
那怎么办?
思路可能是多少步保存一下模型参数,
即不能训练结束之后才保存
一般目录可能会出错的有
最后要以/结尾
比如上面如果是./ckpt_dir就会报错
还有/等价于\
还有.表示当前目录
os.path.join函数
format函数
程序中是每个5轮保存一次,保存的时候设置了序号,这样方便查阅,一次保存是保存三个文件
程序中是每个5轮保存一次,保存的时候设置了序号,这样方便查阅,一次保存是保存三个文件
程序中是每个5轮保存一次,保存的时候设置了序号,这样方便查阅,一次保存是保存三个文件
checkpoint只有一份
那这样的话会不会保存很多文件,
答案是不会
因为模型保存机制缺省(默认)的是保存最近5份
听说还可以在导入模型后进行断点续训
听说还可以在导入模型后进行断点续训
听说还可以在导入模型后进行断点续训
这个要到后面才会用到