RuntimeError: Error(s) in loading state_dict for YOLOX:
1. 目标检测类别不一致 ，即训练权重是的类别文件和使用的模型类别文件不一致
2. 加载使用模型时和训练模型时的环境不一致，这个模型是用GPU训练的，我本是使用的是CPU，所以会有些问题
3. 在加载模型权重的时候，model.load_state_dict(ckpt) —> model.load_state_dict(ckpt, False)
版权声明：本文为fangweijiex原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。