lvkaokao commited on
Commit
fbbb8d6
1 Parent(s): 292ea08

update hyper-parameters

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -34,8 +34,8 @@ The following hyperparameters were used during training:
34
  - total_train_batch_size: 64
35
  - total_eval_batch_size: 8
36
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
37
- - lr_scheduler_type: linear
38
- - lr_scheduler_warmup_ratio: 0.02
39
  - num_epochs: 2.0
40
 
41
  ## Inference with transformers
 
34
  - total_train_batch_size: 64
35
  - total_eval_batch_size: 8
36
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
37
+ - lr_scheduler_type: cosine
38
+ - lr_scheduler_warmup_ratio: 0.03
39
  - num_epochs: 2.0
40
 
41
  ## Inference with transformers