LukeOLuck commited on
Commit
7873802
1 Parent(s): 92414db

Model save

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -40,10 +40,12 @@ The following hyperparameters were used during training:
40
  - train_batch_size: 32
41
  - eval_batch_size: 8
42
  - seed: 42
 
 
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: constant
45
  - lr_scheduler_warmup_ratio: 0.03
46
- - num_epochs: 2
47
 
48
  ### Training results
49
 
 
40
  - train_batch_size: 32
41
  - eval_batch_size: 8
42
  - seed: 42
43
+ - gradient_accumulation_steps: 2
44
+ - total_train_batch_size: 64
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: constant
47
  - lr_scheduler_warmup_ratio: 0.03
48
+ - num_epochs: 3
49
 
50
  ### Training results
51