yolo5问题汇总

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')