Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -222,7 +222,7 @@ def train_model(model_name, data, batch_size, epochs, learning_rate=1e-5, max_le
|
|
222 |
num_train_epochs=epochs,
|
223 |
per_device_train_batch_size=batch_size,
|
224 |
per_device_eval_batch_size=batch_size,
|
225 |
-
|
226 |
save_strategy='epoch', # Ensure save strategy matches evaluation strategy
|
227 |
learning_rate=learning_rate,
|
228 |
save_steps=500,
|
|
|
222 |
num_train_epochs=epochs,
|
223 |
per_device_train_batch_size=batch_size,
|
224 |
per_device_eval_batch_size=batch_size,
|
225 |
+
eval_strategy='epoch', # Set evaluation strategy
|
226 |
save_strategy='epoch', # Ensure save strategy matches evaluation strategy
|
227 |
learning_rate=learning_rate,
|
228 |
save_steps=500,
|