Spaces:

mojad121
/

Chatbot-By-Mujtaba

Sleeping

mojad121 commited on Jun 22

Commit

cab3e7b

verified ·

1 Parent(s): d3fa3b8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ def fine_tune_model(train_data, test_data):
     training_args = TrainingArguments(
         output_dir="./results",
-        eval_strategy="steps",
         eval_steps=500,
         learning_rate=3e-5,
         per_device_train_batch_size=BATCH_SIZE,

     training_args = TrainingArguments(
         output_dir="./results",
+        evaluation_strategy="steps",
         eval_steps=500,
         learning_rate=3e-5,
         per_device_train_batch_size=BATCH_SIZE,