From a9302b5e748340ff976bc11bbc9c4a9ddd2b77a5 Mon Sep 17 00:00:00 2001 From: The-truthh <821372701@qq.com> Date: Tue, 18 Jun 2024 10:42:31 +0800 Subject: [PATCH] fix: fix bugs of repvgg --- configs/repvgg/repvgg_a0_ascend.yaml | 3 +-- configs/repvgg/repvgg_a1_ascend.yaml | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) 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