Update schedule_adamw_cos_10e.py

Mountchicken · Aug 7, 2023 · 3b49deb · 3b49deb
1 parent 36ff31e
commit 3b49deb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmocr-dev-1.x/configs/textrecog/_base_/schedules/schedule_adamw_cos_10e.py b/mmocr-dev-1.x/configs/textrecog/_base_/schedules/schedule_adamw_cos_10e.py
@@ -3,7 +3,7 @@
     type='OptimWrapper',
     optimizer=dict(
         type='AdamW',
-        lr=4e-4,
+        lr=1e-4,
         betas=(0.9, 0.999),
         eps=1e-08,
         weight_decay=0.01))
@@ -16,6 +16,6 @@
     dict(
         type='CosineAnnealingLR',
         T_max=10,
-        eta_min=4e-6,
+        eta_min=1e-8,
         convert_to_iter_based=True)
 ]