UserWarning:火炬。meshgrid:在以后的版本中,它将被要求通过索引ing argu

The following error occurred when doing yolov5 locally (no gpu locally)

After consulting the data direction, the memory is not enough. In this case, the batchsize can be reduced.
Originally:
python train.py --img 640 --batch 16 --epochs 5
Now:
python train.py --img 640 --batch 4 --epochs 5
p>