实操教程｜PyTorch实现断点继续训练

文章预览

点击上方 “ 小白学视觉 ”，选择加" 星标 "或“ 置顶 ” 重磅干货，第一时间送达 导读   本文整理了pytorch实现断电继续训练时需要注意的要点，附有代码详解。 最近在尝试用CIFAR10训练分类问题的时候，由于数据集体量比较大，训练的过程中时间比较长，有时候想给停下来，但是停下来了之后就得重新训练，之前师兄让我们学习断点继续训练及继续训练的时候注意epoch的改变等，今天上午给大致整理了一下，不全面仅供参考 Epoch: 9 | train loss: 0.3517 | test accuracy: 0 . 7184 | train time: 14215.1018 s Epoch: 9 | train loss: 0 . 2471 | test accuracy: 0.7252 | train time: 14309.1216 s Epoch: 9 | train loss: 0.4335 | test accuracy: 0 . 7201 | train time: 14403.2398 s Epoch: 9 | train loss: 0 . 2186 | test accuracy: 0.7242 | train time: 14497.1921 s Epoch: 9 | train loss: 0.2127 | test accurac ………………………………