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

main
songhyeonsoo 1 month ago
parent 7c58619098
commit 429f794018

@ -1,6 +1,6 @@
Global: Global:
use_gpu: True use_gpu: True
epoch_num: 50 epoch_num: 100
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: ./output/kr_lp_pgnet/ save_model_dir: ./output/kr_lp_pgnet/
@ -53,7 +53,7 @@ Optimizer:
lr: lr:
name: Cosine name: Cosine
learning_rate: 0.001 learning_rate: 0.001
warmup_epoch: 50 warmup_epoch: 5
regularizer: regularizer:
name: 'L2' name: 'L2'
factor: 0.0001 factor: 0.0001

Loading…
Cancel
Save