switch learning_rate and lr

2026-09-24 23:33:08 +08:00 · 2020-11-05 20:47:16 +08:00
parent 41b33c9e9c
commit d092a5a22f
2 changed files with 4 additions and 4 deletions
@@ -44,9 +44,9 @@ Optimizer:
  name: Adam
  beta1: 0.9
  beta2: 0.999
-  learning_rate:
+  lr:
 #    name: Cosine
-    lr: 0.001
+    learning_rate: 0.001
 #    warmup_epoch: 0
  regularizer:
    name: 'L2'
@@ -29,8 +29,8 @@ Optimizer:
  name: Adam
  beta1: 0.9
  beta2: 0.999
-  learning_rate:
-    lr: 0.0005
+  lr:
+    learning_rate: 0.0005
  regularizer:
    name: 'L2'
    factor: 0.00001