Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -80,7 +80,7 @@ def fine_tune_model(tokenized_dataset):
|
|
80 |
|
81 |
training_args = TrainingArguments(
|
82 |
output_dir="./results",
|
83 |
-
|
84 |
learning_rate=5e-5,
|
85 |
per_device_train_batch_size=4,
|
86 |
per_device_eval_batch_size=4,
|
|
|
80 |
|
81 |
training_args = TrainingArguments(
|
82 |
output_dir="./results",
|
83 |
+
evaluation_strategy="epoch",
|
84 |
learning_rate=5e-5,
|
85 |
per_device_train_batch_size=4,
|
86 |
per_device_eval_batch_size=4,
|