AttributeError: ‘ConfigDict‘ object has no attribute ‘device‘
AttributeError: 'ConfigDict' object has no attribute 'device'
·
mmdetection训练自己的coco数据集通过命令行运行tools/train.py文件时,报错:AttributeError: ‘ConfigDict’ object has no attribute ‘device’
在GitHub找到了解决方法:
在tools/train.py的meta['config'] = cfg.pretty_text
下面添加如下代码:
cfg.device='cuda'
更多推荐
已为社区贡献1条内容
所有评论(0)