pytorch模型部署去除训练信息方法

使用pytorch框架进行模型训练的时候，尤其需要进行断点续接训练的时候，我们需要对模型权重中的epoch信息进行去除，从而实现模型的进一步训练。

–epoch 20的模型训练得到的.pt文件
使用指令：

python3 train.py --epoch 20 --weight last.pt

模型训练报错，原因很简单，.pt文件文件中包含有epoch的训练信息，需要进行清除，也可以使用大于last.pt的训练次数

清除epoch信息指令：

python -c "from models import *; convert('cfg/yolov3.cfg','weight/last.pt')"

完成了，就可以清除权重中的epoch信息，重新进行模型训练