acecalisto3 commited on
Commit
fbd4bb6
·
verified ·
1 Parent(s): 8f23203

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- evaluation_strategy='epoch', # Set evaluation strategy
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,