训练yolov5模型时遇到：TypeError: ‘NoneType‘ object is not callable

Posted 2021-09-22 Rainbowman 0

刚刚在用矩池云的GPU训练yolov5模型（Google Colab的免费GPU确实不错但还是有些慢），同样的操作在Google Colab上就能跑，在矩池云这里用命令：

python train.py --img 640 --batch 64 --epochs 180 --data /mnt/yolov5-master/traindata/traindata.yaml --weights yolov5s.pt --nosave --cache

就会出现报错：

捣鼓了半个小时才发现原来是batch设置大了（我猜是显存不够了），把batch设置小一些就可以正常训练了，比如设置成–batch 32或者更小。

这样就可以正常训练了 ：）