模型的保存与还原

有没有想过一个需要训练一个礼拜的模型 到最后一天突然断电了 那怎么办? 思路可能是多少步保存一下模型参数, 即不能训练结束之后才保存 一般目录可能会出错的有 最后要以/结尾 比如上面如果是./ckpt_dir就会报错 还有/等价于\ 还有.表示当前目录 os.path.join函数 format函数 程序中是每个5轮保存一次,保存的时候设置了序号,这样方便查阅,一次保存是保存三个文件 程序中是每个
相关文章
相关标签/搜索