Youliang commited on
Commit
3c953fa
·
verified ·
1 Parent(s): f8fec31

Update README.md

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