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

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐