Reyad-Ahmmed commited on
Commit
698106d
·
verified ·
1 Parent(s): b497dbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -111,7 +111,8 @@ if (runModel=='1'):
111
  weight_decay=0.02,
112
  logging_dir='./logs_' + modelNameToUse,
113
  logging_steps=10,
114
- evaluation_strategy="epoch",
 
115
  load_best_model_at_end=True, # Load the best model based on evaluation
116
  )
117
 
 
111
  weight_decay=0.02,
112
  logging_dir='./logs_' + modelNameToUse,
113
  logging_steps=10,
114
+ evaluation_strategy="epoch", # Evaluation strategy is 'epoch'
115
+ save_strategy="epoch", # Save strategy should also be 'epoch'
116
  load_best_model_at_end=True, # Load the best model based on evaluation
117
  )
118