mmdetection训练自己的coco数据集通过命令行运行tools/train.py文件时,报错:AttributeError: ‘ConfigDict’ object has no attribute ‘device’
在GitHub找到了解决方法:
在tools/train.py的meta['config'] = cfg.pretty_text下面添加如下代码:

cfg.device='cuda'
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐