[Fix] Fix mix training on Ascend NPU

open-mmlab · Jul 21, 2023 · eb1ece7 · eb1ece7
1 parent 0beaf69
commit eb1ece7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mmseg/apis/train.py b/mmseg/apis/train.py
@@ -137,9 +137,9 @@ def train_segmentor(model,
             meta=meta))
 
     if cfg.device == 'npu' and not is_npu_support_full_precision():
-        optimiter_config = dict(type='Fp16OptimizerHook', loss_scale='dynamic')
-        cfg.optimizer_config = optimiter_config if \
-            not cfg.optimizer_config else cfg.optimizer_config
+        cfg.optimizer_config = cfg.optimizer_config or {}
+        cfg.optimizer_config['type'] = 'Fp16OptimizerHook'
+        cfg.optimizer_config['loss_scale'] = 'dynamic'
 
     # register hooks
     runner.register_training_hooks(cfg.lr_config, cfg.optimizer_config,