config: extend training to 100 epochs, reduce warmup to 5 for faster lr rise

1 month ago · 429f794018
parent 7c58619098
commit 429f794018
1 changed files with 2 additions and 2 deletions
--- a/configs/kr_lp_pgnet.yml
+++ b/configs/kr_lp_pgnet.yml
@ -1,6 +1,6 @@
 Global:
  use_gpu: True
-  epoch_num: 50
+  epoch_num: 100
  log_smooth_window: 20
  print_batch_step: 10
  save_model_dir: ./output/kr_lp_pgnet/
@ -53,7 +53,7 @@ Optimizer:
  lr:
    name: Cosine
    learning_rate: 0.001
-    warmup_epoch: 50
+    warmup_epoch: 5
  regularizer:
    name: 'L2'
    factor: 0.0001