mmdetection使用yolox训练时出现find_unused_parameters报错

2022年06月10日09:54:00 更新

——————————————————————————————

问题在于没有修改类别数

————————————————————————————————

github上已经上报了该问题：https://github.com/open-mmlab/mmdetection/issues/7298

问题描述：

我使用2张2080ti的卡训练，在epoch1中迭代一些次数后就卡着不动了，但是gpu利用率还在浮动；然后我使用一张卡训练，就报出该issue下的问题。

解决办法：

我检查了我的所有图片，每张图片都有框，框位置正常，然后我使用报错提示的尝试方法find_unused_parameters=True（在mmdet/apis/train.py的if distributed下加入），两张卡下目前训练2个epoch没有报错，继续观察中