Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -53,14 +53,14 @@ max_steps = steps_per_epoch * num_train_epochs
|
|
53 |
|
54 |
# Define training arguments
|
55 |
training_args = TrainingArguments(
|
56 |
-
output_dir='
|
57 |
-
|
58 |
learning_rate=2e-5,
|
59 |
per_device_train_batch_size=8, # You can adjust batch size here
|
60 |
per_device_eval_batch_size=8,
|
61 |
num_train_epochs=num_train_epochs, # Setting epochs to 3
|
62 |
weight_decay=0.01,
|
63 |
-
logging_dir='
|
64 |
logging_steps=500,
|
65 |
save_steps=1000,
|
66 |
load_best_model_at_end=True,
|
|
|
53 |
|
54 |
# Define training arguments
|
55 |
training_args = TrainingArguments(
|
56 |
+
output_dir='results',
|
57 |
+
eval_strategy='epoch', # Evaluate at the end of each epoch
|
58 |
learning_rate=2e-5,
|
59 |
per_device_train_batch_size=8, # You can adjust batch size here
|
60 |
per_device_eval_batch_size=8,
|
61 |
num_train_epochs=num_train_epochs, # Setting epochs to 3
|
62 |
weight_decay=0.01,
|
63 |
+
logging_dir='logs',
|
64 |
logging_steps=500,
|
65 |
save_steps=1000,
|
66 |
load_best_model_at_end=True,
|