算法可正常训练:环境window10+pytorch1.8.0+cuda11.1+detectron2-0.2.1
a)detectron2-0.2.1配置参考:windows10系统安装detectron2_会飞的小菜鸡的博客-CSDN博客_detectron2 安装
其他版本的detectron2(>=0.2.1)也可以。
问题汇总:
1、BrokenPipeError: [Errno 32] Broken pipe
这个问题我直接将torch.utils.data.dataloader lines:185 中的self.num_workers强制设置成1
2、FloatingPointError: Predicted boxes or scores contain Inf/NaN. Training has diverged.
batch_size调大解决此问题。