Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ tokenized_dataset = dataset.map(tokenize_function, batched=True)
|
|
20 |
training_args = TrainingArguments(
|
21 |
output_dir="./results",
|
22 |
overwrite_output_dir=True,
|
23 |
-
|
24 |
learning_rate=5e-5,
|
25 |
per_device_train_batch_size=2,
|
26 |
num_train_epochs=3,
|
|
|
20 |
training_args = TrainingArguments(
|
21 |
output_dir="./results",
|
22 |
overwrite_output_dir=True,
|
23 |
+
eval_strategy="epoch", # Updated
|
24 |
learning_rate=5e-5,
|
25 |
per_device_train_batch_size=2,
|
26 |
num_train_epochs=3,
|