经过固定次数的迭代训练后，有什么方法可以在YOLOv5中备份权重？

Question

我想在训练YOLOv5时备份重量，这样，如果由于某种原因训练停止，我不必像yolov3一样重新开始，您可以提及每100次迭代后就在其中备份重量的文件夹

Answer 1

在“运行”文件夹中，您可以在名为项目的文件夹下找到重量文件（* .pt）。

例如，

保存pt文件，并在检测到这种情况时加载一个文件。

!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}"