pytorch教程:save and load

torch有一个“save and load”机制,即训练意外终止了,可以保存训练的中间文件(ckpt.mdl),然后恢复训练的最开始重新加载进来。
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_40199232/article/details/108702012