Is there any way to backup weights in YOLOv5 when training after a fixed number of iterations?

Question

I want to backup weights while training YOLOv5 so that if the training stops for some reason i dont have to start all over again like in yolov3 you can mention the folder in which weights are backed up after every 100 iterations

https://github.com/ultralytics/yolov5 this is the repository I cloned it from — Alrik Fernandes, Jul 02 '20 at 04:09

score 0 · Answer 1 · answered Sep 22 '20 at 03:22

In 'runs' folder, you can find weights file (*.pt) under the folder named your project.

For example,

Save the pt file, and load one when you detect something like this.

!python detect.py --weights '/content/drive/My Drive/yolov5/runs/exp8_i2_PA+pano_yolov5x_results/weights/best_i2_PA+pano_yolov5x_results.pt' --img 512 --conf 0.5 --source "{test_img_path}"

Is there any way to backup weights in YOLOv5 when training after a fixed number of iterations?

1 Answers1