Co-DETR icon indicating copy to clipboard operation
Co-DETR copied to clipboard

fp16 setting

Open Bailin-He opened this issue 1 year ago • 3 comments

Traceback (most recent call last): File "tools/train.py", line 245, in main() File "tools/train.py", line 241, in main meta=meta) File "/mnt/data/Co-DETR/tools/mmdet/apis/train.py", line 187, in train_detector elif distributed and 'type' not in cfg.optimizer_config: File "/mnt/data/miniconda/envs/co-detr/lib/python3.7/site-packages/mmcv/utils/config.py", line 513, in getattr return getattr(self._cfg_dict, name) File "/mnt/data/miniconda/envs/co-detr/lib/python3.7/site-packages/mmcv/utils/config.py", line 49, in getattr raise ex AttributeError: 'ConfigDict' object has no attribute 'optimizer_config'

Bailin-He avatar Aug 11 '24 10:08 Bailin-He

似乎是配置文件中没有设置optimizer_config导致的

Bailin-He avatar Aug 11 '24 12:08 Bailin-He

This repository does not support fp16 training.

TempleX98 avatar Aug 11 '24 16:08 TempleX98

This repository does not support fp16 training.

mmdet.apis.train.py的182行 `

fp16 setting

fp16_cfg = cfg.get('fp16', None)
if fp16_cfg is not None:
    optimizer_config = Fp16OptimizerHook(
        **cfg.optimizer_config, **fp16_cfg, distributed=distributed)
elif distributed and 'type' not in cfg.optimizer_config:
    optimizer_config = OptimizerHook(**cfg.optimizer_config)
else:
    optimizer_config = cfg.optimizer_config

` 但是配置文件里面只有optimizer,没有optimizer_config,所以报的这个错

Bailin-He avatar Aug 11 '24 17:08 Bailin-He