YOLO模型训练进度卡在0%

YOLO模型训练进度卡在0%

  • 问题描述
  • 解决方案

问题描述

使用train.py训练模型,epoch卡在第一轮的0%

解决方案

检查数据集
特别是XXX.yaml文件和classes.txt文件类型是否匹配

你可能感兴趣的:(pytorch,深度学习)