Reyad-Ahmmed commited on
Commit
f3f2742
·
verified ·
1 Parent(s): 9bb3f13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -100,7 +100,7 @@ if (runModel=='1'):
100
  # Create an instance of the custom loss function
101
  training_args = TrainingArguments(
102
  output_dir='./results_' + modelNameToUse,
103
- num_train_epochs=25,
104
  per_device_train_batch_size=2,
105
  per_device_eval_batch_size=2,
106
  warmup_steps=500,
 
100
  # Create an instance of the custom loss function
101
  training_args = TrainingArguments(
102
  output_dir='./results_' + modelNameToUse,
103
+ num_train_epochs=4,
104
  per_device_train_batch_size=2,
105
  per_device_eval_batch_size=2,
106
  warmup_steps=500,