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,
|