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