yolov5训练自己的数据集train.py

Question

我用yolov5训练自己的数据集，有1470张图片，用的预训练模型是yolov5s.pt,然后在命令行输入了

python train.py --img 640 --batch 4 --epoch 100 --data ./data/myvoc.yaml --cfg ./models/yolov5s.yaml --weights ./weights/yolov5s.pt --device 0 --workers 0

结果一直卡在这个界面

ctrl z或者ctrl c都停不了。
我的环境是windows+(torch1.7.1+cu101)+(torchaudio0.10.1+cu111)+(torchvision0.8.2+cu101)+CUDA11.3+cudnn7.6.0.4

Answer 1

你这槽点不是一般的多，cuda要和前面的匹配，前面的cuda版本号要一致