genne commited on
Commit
1da6351
1 Parent(s): 28025a2

Model save

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -39,12 +39,12 @@ The following hyperparameters were used during training:
39
  - eval_batch_size: 8
40
  - seed: 42
41
  - distributed_type: multi-GPU
42
- - num_devices: 32
43
- - total_train_batch_size: 256
44
- - total_eval_batch_size: 256
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
- - num_epochs: 10.0
48
 
49
  ### Training results
50
 
 
39
  - eval_batch_size: 8
40
  - seed: 42
41
  - distributed_type: multi-GPU
42
+ - num_devices: 8
43
+ - total_train_batch_size: 64
44
+ - total_eval_batch_size: 64
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
+ - num_epochs: 1.0
48
 
49
  ### Training results
50