UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the index ing argu

经查阅资料方向是内存不够，遇到这种情况可以将batchsize减小。在本地做yolov5时出现了以下错误

woshicaiji12138

8666人浏览 · 2022-07-29 04:05:19

woshicaiji12138 · 2022-07-29 04:05:19 发布

在本地做yolov5时出现了以下错误（本地无gpu)
在这里插入图片描述
经查阅资料方向是内存不够，遇到这种情况可以将batchsize减小。
原来：
python train.py --img 640 --batch 16 --epochs 5
现在：
python train.py --img 640 --batch 4 --epochs 5