msmaje commited on
Commit
5b47a77
·
verified ·
1 Parent(s): a19d57d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -343,7 +343,7 @@ def train_model_inline(uploaded_file, text_column, label_column, num_epochs, bat
343
  learning_rate=learning_rate,
344
  logging_dir=str(output_dir / "logs"),
345
  logging_steps=logging_steps,
346
- evaluation_strategy="steps", # Corrected parameter name
347
  eval_steps=eval_steps,
348
  save_steps=save_steps,
349
  save_total_limit=2,
 
343
  learning_rate=learning_rate,
344
  logging_dir=str(output_dir / "logs"),
345
  logging_steps=logging_steps,
346
+ eval_strategy="steps", # Corrected back to eval_strategy
347
  eval_steps=eval_steps,
348
  save_steps=save_steps,
349
  save_total_limit=2,