觉得当前训练的epoch数量还不够,想要再多训练几个epoch,又不想从头开始训练,那下面的方法很适合你:
-
1. 修改train.py
parser.add_argument( "--resume", default=True, action="store_true", help="resume training" #改为True ) parser.add_argument("-c", "--ckpt", default='YOLOX_outputs/yolox_voc_s/latest_ckpt.pth', type=str, help="checkpoint file") #预训练权重用最后一个epoch的权重 parser.add_argument( "-e", "--start_epoch", default=100, #从第100epoch开始,根据自己的实际情况修改 type=int, help="resume training start epoch", )
-
2. 也可直接在命令行操作:
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 -c YOLOX_outputs/yolox_voc_s/latest_ckpt.pth.tar -resume -start_epoch=100
-
再次训练,就可以发现你的epoch不是从1开始了