Youliang commited on
Commit
41db283
·
verified ·
1 Parent(s): c8cb8d7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -34,14 +34,14 @@ More information needed
34
 
35
  The following hyperparameters were used during training:
36
  - learning_rate: 0.0001
37
- - train_batch_size: 16
38
  - eval_batch_size: 1
39
  - seed: 1
40
  - distributed_type: multi-GPU
41
  - num_devices: 8
42
  - gradient_accumulation_steps: 2
43
  - total_train_batch_size: 128
44
- - total_eval_batch_size: 6
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: cosine
47
  - num_epochs: 2.0
 
34
 
35
  The following hyperparameters were used during training:
36
  - learning_rate: 0.0001
37
+ - train_batch_size: 8
38
  - eval_batch_size: 1
39
  - seed: 1
40
  - distributed_type: multi-GPU
41
  - num_devices: 8
42
  - gradient_accumulation_steps: 2
43
  - total_train_batch_size: 128
44
+ - total_eval_batch_size: 8
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: cosine
47
  - num_epochs: 2.0