nroggendorff commited on
Commit
2ee4483
·
verified ·
1 Parent(s): 756483c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -100,6 +100,7 @@ args = TrainingArguments(
100
  output_dir="mayo",
101
  num_train_epochs=2,
102
  per_device_train_batch_size=128,
 
103
  learning_rate=1e-5,
104
  save_steps=100000,
105
  fp16=True,
 
100
  output_dir="mayo",
101
  num_train_epochs=2,
102
  per_device_train_batch_size=128,
103
+ gradient_accumulation_steps=8,
104
  learning_rate=1e-5,
105
  save_steps=100000,
106
  fp16=True,