PETR_code_note
PETR code note
cfg = Config.fromfile(args.config)
#从args更新读取的config文件,args优先级>cfg优先级,args定义了cfg文件中没有定义的work_dir等参数,还有一部分需要覆盖cfg的参数
cfg.merge_from_dict()
合并字典到配置 通过
cfg.merge_from_dict
函数接口可以实现对字典内容进行合并,典型用法如下:= osp.join(data_path, 'config/a.py') cfg_file = Config.fromfile(cfg_file) cfg = {'item2.a': 1, 'item2.b': 0.1, 'item3': False} input_options cfg.merge_from_dict(input_options) # 原始 a.py 内容为: = [1, 2] item1 = {'a': 0} item2 = True item3 = 'test' item4 # 进行合并后, cfg 内容 = [1, 2] item1 = dict(a=1, b=0.1) item2 = False item3 = 'test' item4
```python if cfg.plugin: import importlib
#将plugin批量导入模型环境
#plugin_dir='projects/mmdet3d_plugin/'
torch.backends.cudnn.benchmark = True
对模型里的卷积层进行预先的优化,也就是在每一个卷积层中测试 cuDNN 提供的所有卷积实现算法,然后选择最快的那个。这样在模型启动的时候,只要额外多花一点点预处理时间,就可以较大幅度地减少训练时间。
All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.