yolov5训练中的一些问题

一、模型大小的控制

yolov5在模型的yaml文件中有两个指标是控制模型深度和宽度的,

depth_multiple: 0.33  # model depth multiple
width_multiple: 0.50  # layer channel multiple

调节这两个值可以控制模型大小:
1、将s网络的指标分别设置如下,参数量不同

yolov5训练中的一些问题_第1张图片
yolov5训练中的一些问题_第2张图片
2、将s和x网络的指标设置一样后,参数量是相同的
yolov5训练中的一些问题_第3张图片
yolov5训练中的一些问题_第4张图片

二、训练过程中总是出现CUDA out of memory问题

报错如下:

RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 6.00 GiB total capacity; 1.06 GiB already allocated; 2.75 GiB free; 

都说是batch_size太大,但是我batch_size是1,是不能再小了,何况我batch_size是16的时候照样跑,但是它就是报这个错,在我千帆寻找下,发现--------------只需要重启电脑就可以啦!!!!

你可能感兴趣的:(深度学习,人工智能)