diff --git a/configs/repvgg/repvgg_a0_ascend.yaml b/configs/repvgg/repvgg_a0_ascend.yaml index 534d9d1a7..fd7370628 100644 --- a/configs/repvgg/repvgg_a0_ascend.yaml +++ b/configs/repvgg/repvgg_a0_ascend.yaml @@ -45,7 +45,6 @@ decay_epochs: 390 opt: "momentum" momentum: 0.9 weight_decay: 0.0001 -loss_scale_type: "dynamic" -drop_overflow_update: True +loss_scale: 1024 use_nesterov: False eps: 0.00000001 diff --git a/configs/repvgg/repvgg_a1_ascend.yaml b/configs/repvgg/repvgg_a1_ascend.yaml index 3144967f0..1945fb70c 100644 --- a/configs/repvgg/repvgg_a1_ascend.yaml +++ b/configs/repvgg/repvgg_a1_ascend.yaml @@ -45,7 +45,6 @@ decay_epochs: 235 opt: "momentum" momentum: 0.9 weight_decay: 0.0001 -loss_scale_type: "dynamic" -drop_overflow_update: True +loss_scale: 1024 use_nesterov: False eps: 0.00000001