lvkaokao commited on
Commit
0ff1545
1 Parent(s): 342b16d

update train args.

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -28,9 +28,9 @@ The following hyperparameters were used during training:
28
  - train_batch_size: 1
29
  - eval_batch_size: 2
30
  - seed: 42
31
- - distributed_type: multi-GPU
32
- - num_devices: 4
33
- - gradient_accumulation_steps: 16
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
 
28
  - train_batch_size: 1
29
  - eval_batch_size: 2
30
  - seed: 42
31
+ - distributed_type: multi-HPU
32
+ - num_devices: 8
33
+ - gradient_accumulation_steps: 8
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