模型的保存與還原

有沒有想過一個需要訓練一個禮拜的模型 到最後一天突然斷電了 那怎麼辦? 思路可能是多少步保存一下模型參數, 即不能訓練結束之後才保存 一般目錄可能會出錯的有 最後要以/結尾 比如上面如果是./ckpt_dir就會報錯 還有/等價於\ 還有.表示當前目錄 os.path.join函數 format函數 程序中是每個5輪保存一次,保存的時候設置了序號,這樣方便查閱,一次保存是保存三個文件 程序中是每個
相關文章
相關標籤/搜索