Spaces:

Reyad-Ahmmed
/

HF_Python

Runtime error

Reyad-Ahmmed commited on Jan 3

Commit

9015400

verified ·

1 Parent(s): 0265174

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,7 +104,7 @@ if (runModel=='1'):
     # Create an instance of the custom loss function
     training_args = TrainingArguments(
         output_dir='./results_' + modelNameToUse,
-        num_train_epochs=5,
         per_device_train_batch_size=8,
         per_device_eval_batch_size=8,
         warmup_steps=500,

     # Create an instance of the custom loss function
     training_args = TrainingArguments(
         output_dir='./results_' + modelNameToUse,
+        num_train_epochs=15,
         per_device_train_batch_size=8,
         per_device_eval_batch_size=8,
         warmup_steps=500,