torch有一个“save and load”机制,即训练意外终止了,可以保存训练的中间文件(ckpt.mdl),然后恢复训练的最开始重新加载进来。
pytorch教程:save and load
猜你喜欢
转载自blog.csdn.net/qq_40199232/article/details/108702012
今日推荐
周排行