神经网络训练中resume的使用_代码resume功能

在训练神经网络的过程中，有时候需要需要训练几千个Epoch，比如Polylanenet车道线检测模型就需要训练2695个Epoch，训练的过程中如果因为停电等原因被迫退出训练是很遭心的，又得重头开始训练，很浪费时间，好在大多数的神经网络模型给我们提供了resume选项，这个选项的意思是是否恢复训练，利用好这个命令，我们就不用从头开始训练了，节约了时间，下面看看如何让使用resume选项的吧：

第一步：首先找到训练的配置文件，我这里以ultra-fast-lane-detection模型的tusimple.py文件为例，打开tusimple.py，找到resume选项的位置：

 可以看到，resume=None，这说明还没告诉模型从哪里开始训练，现在我们告述它一个开始训练的位置，我这里以下图所示中训练生成的模型（黄框）开始

把上述黄框模型的绝对路径放到tusimple.py文件中的resume处， 如下图所示：

 然后开始训练，可以看到，训练的信息出现了resume model from........

 我这里训练三轮，可以看到保存训练模型的日志文件信息如下：

 可以看到新生成的训练模型不是从ep000.pth开始，而是接着上面ep002.pth从ep003.pth开始的，其他的神经网络模型的resume使用方法类似，使用resume选项，训练的时候可以节约时间，希望以上内容对宝子们有所帮助！