Reyad-Ahmmed commited on
Commit
5e1124f
·
verified ·
1 Parent(s): f8beb3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -105,8 +105,8 @@ if (runModel=='1'):
105
  training_args = TrainingArguments(
106
  output_dir='./results_' + modelNameToUse,
107
  num_train_epochs=5,
108
- per_device_train_batch_size=5,
109
- per_device_eval_batch_size=5,
110
  warmup_steps=500,
111
  weight_decay=0.02,
112
  logging_dir='./logs_' + modelNameToUse,
 
105
  training_args = TrainingArguments(
106
  output_dir='./results_' + modelNameToUse,
107
  num_train_epochs=5,
108
+ per_device_train_batch_size=8,
109
+ per_device_eval_batch_size=8,
110
  warmup_steps=500,
111
  weight_decay=0.02,
112
  logging_dir='./logs_' + modelNameToUse,