alvanlii commited on
Commit
0d7d99f
·
1 Parent(s): 97aa366

Update batch size

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ For training, three datasets were used:
42
  - train_batch_size: 25 (on 2 GPUs)
43
  - eval_batch_size: 8
44
  - gradient_accumulation_steps: 2
45
- - total_train_batch_size: 16x2x2=64
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 500
 
42
  - train_batch_size: 25 (on 2 GPUs)
43
  - eval_batch_size: 8
44
  - gradient_accumulation_steps: 2
45
+ - total_train_batch_size: 25x2x2=100
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 500