yolov5-resnet训练问题解决

1.首先安装torch、torchvision、progress、matplotlib

pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install progress -i https://pypi.tuna.tsinghua.edu.cn/simple

2.将cifar.py下的240行的asyncTrue删除

3.出现下面爆红，torchvision版本不匹配，升级torchvision版本

pip install --upgrade torchvision

4.训练TRAINING.MD

5.如果没有安装cuda,就会报下面这个

点击链接，将cuda改为cpu

5.将view改为reshape，继续训练TRAINING.md

6.出现以下报错，将.data[0],[0]改为.item()

7.最后再运行TRAINING.md

选用自己想用的模型