为什么我的yolov5s跑的很慢

Question

 Epoch    GPU_mem   box_loss   obj_loss   cls_loss  Instances       Size
      18/99      6.57G    0.04504    0.03544   0.008159         65        640: 100%|██████████| 7/7 [00:54<00:00,  7.79s/it]
                 Class     Images  Instances          P          R      mAP50   mAP50-95: 100%|██████████| 4/4 [01:27<00:00, 21.85s/it]
                   all        200        993      0.819       0.94      0.959      0.587

      Epoch    GPU_mem   box_loss   obj_loss   cls_loss  Instances       Size
      19/99       7.8G    0.03693    0.03706   0.007061         76        640: 100%|██████████| 7/7 [00:24<00:00,  3.55s/it]
                 Class     Images  Instances          P          R      mAP50   mAP50-95: 100%|██████████| 4/4 [01:53<00:00, 28.31s/it]
                   all        200        993      0.923      0.941      0.978      0.509

别人的单位都是it/s,而我的是it/s，不理解为什么
补充一下我的设备情况，4060显卡， 16GB内存，
满血i9英特尔芯片，8核处理器
下面是我在train里面调整的参数：

    parser.add_argument("--batch-size", type=int, default=24, help="total batch size for all GPUs, -1 for autobatch")
    workspace为8

补充：别人3060跑相同的数据集一共只需要十几分钟，而4060却需要这么长时间，我不是很理解
有的时候快，有的时候慢，但总体上偏慢，大概几分钟乃至十几分钟才能跑完一次Epoch

Answer 1

晚上好🌙🌙🌙
本答案参考ChatGPT-3.5

您好，yolov5s跑的速度慢可能是以下几个因素导致的：

1. GPU性能不足：yolov5s模型是一个比较大的模型，需要较快的GPU来提供足够的计算力。您目前使用的是RTX 4060显卡，可能有些不足。 **解决方案：**尝试使用其他性能更佳的GPU，如V100、A100等。

2. 输入图片大小：如果您的输入图片过大，那么就会影响模型运算速度。您可以尝试将输入图片大小缩小一些。 **解决方案：**通过修改yolov5.yaml文件里的img参数，将输入图片大小缩小一些。

3. batch-size太大：如果batch-size太大，那么GPU内存容量可能不足以支撑一次性计算全部数据。 **解决方案：**把batch-size调小一些，或者尝试使用分布式计算来提升速度。

4. 其他因素：可能还包括服务器性能、磁盘读写速度、训练数据集质量等等。

另外，你提供的训练结果中有一项是"Instances"，这个指的是每个epoch的训练图片数量，不是训练的速度。而训练速度(units/sec或iterations/sec)正常情况下是每秒钟处理的图片数量。所以你不用担心单位不一样为什么我的速度慢。

最后，建议您仔细观察训练时GPU使用情况和训练日志，找到瓶颈所在，然后针对性的解决所遇到的问题。