在本地做yolov5时出现了以下错误(本地无gpu)
经查阅资料方向是内存不够,遇到这种情况可以将batchsize减小。
原来:
python train.py --img 640 --batch 16 --epochs 5
现在:
python train.py --img 640 --batch 4 --epochs 5
UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the index ing argu
最新推荐文章于 2024-03-25 09:50:59 发布