1,权重文件无法加载
加载yolov5中的yolov5m.pt时出现下面的错误提示:
AttributeError: ‘Detect’ object has no attribute ‘m’
解决方法,去官网重新下载权重文件
百度云链接:https://pan.baidu.com/s/12tgmPTOglhAKC9_fcN73Tg
提取码:65kl
2,Dataloader出现问题
运行train.py时出现下面的错误提示:
RuntimeError: DataLoader worker (pid(s) 6000) exited unexpectedly
解决方法:
# default原来为8,减小该值
parser.add_argument('--workers', type=int, default=0, help='maximum number of dataloader workers')
3,空间不足
运行train.py时出现下面错误提示:
buy more ram
解决方法:
# 训练总轮次,1个epoch等于使用训练集中的全部样本训练一次,将default值改小一些
parser.add_argument('--epochs', type=int, default=10)
# 批次大小,一次训练所选取的样本数,也改小一些
parser.add_argument('--batch-size', type=int, default=1, help='total batch size for all GPUs')