yolov5在模型的yaml文件中有两个指标是控制模型深度和宽度的,
depth_multiple: 0.33 # model depth multiple width_multiple: 0.50 # layer channel multiple
调节这两个值可以控制模型大小:
1、将s网络的指标分别设置如下,参数量不同
2、将s和x网络的指标设置一样后,参数量是相同的
报错如下:
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的时候照样跑,但是它就是报这个错,在我千帆寻找下,发现--------------只需要重启电脑就可以啦!!!!



