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

在本地做yolov5时出现了以下错误(本地无gpu)
UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the index ing argu_第1张图片
经查阅资料方向是内存不够,遇到这种情况可以将batchsize减小。
原来:
python train.py --img 640 --batch 16 --epochs 5
现在:
python train.py --img 640 --batch 4 --epochs 5

你可能感兴趣的:(常见错误,python,深度学习,开发语言)