我已使用自定义数据启动Yolo 5培训
我使用的命令：
!python train.py --img 640 --batch-size 32 --epochs 5 --data /content/drive/MyDrive/yolov5_dataset/dataset_Trafic/data.yaml --cfg /content/drive/MyDrive/yolov5/models/yolov5s.yaml --name Model培训开始时间和完成时间如下：
为了恢复/继续更多时间，我有以下命令
!python train.py --img 640 --batch-size 32 --epochs 6 --data /content/drive/MyDrive/yolov5_dataset/dataset_Trafic/data.yaml --weights /content/drive/MyDrive/yolov5/runs/train/Model/weights/best.pt --cache --exist-ok
但是培训还是从零开始，如何延续过去的时代。
我还尝试使用resume命令
!python train.py --epochs 10 --resume但我得到以下错误消息
2条答案
按热度按时间omqzjyyz1#
打开
runs/train/Model/opt.yaml文件并更改epochs编号。重试oprakyz72#
尝试以下操作：在runs文件夹（存储先前权重的文件夹）中，找到名为“args.yaml”的文件，将epoch计数增加您现在想要训练模型的epoch数。例如：如果之前您的模型训练了100个历元，那么args.yaml文件将具有一个属性：“epochs：100”假设您要继续训练50个以上的epochs，然后将epochs更改为150，即“epochs：150”保存文件，然后执行训练命令。