YOLOv3_TensorFlow2 使用COCO数据集训练，loss很快变为nan

使用COCO数据集训练，loss很快变为nan

Open Guofeng-Lin opened this issue 4 years ago • 3 comments

启动train_from_scartch.py后，第一个epoch还没有训练完，loss就从61直接变为nan。我没有调整任何超参，是learning rate = 0.001太高了的原因吗？

Oct 09 '21 05:10 Guofeng-Lin

可能是我的代码中COCO数据集处理上的问题

Oct 14 '21 07:10 calmiLovesAI

请问是write_coco_to_txt.py运行过程中出现了问题了吗？

Oct 15 '21 05:10 Guofeng-Lin

应该是解析coco数据集的代码问题，在文件parse_coco.py中，我没有用官方的pycocotools工具

Oct 26 '21 06:10 calmiLovesAI