amanwithaplan commited on
Commit
a21d6c7
·
verified ·
1 Parent(s): 1b22ef9

Eval and save every 25 steps for detailed curves

Browse files
Files changed (1) hide show
  1. train_reranker.py +3 -3
train_reranker.py CHANGED
@@ -295,10 +295,10 @@ def main():
295
  warmup_ratio=0.1,
296
  bf16=True,
297
  eval_strategy="steps",
298
- eval_steps=200,
299
  save_strategy="steps",
300
- save_steps=200,
301
- save_total_limit=2,
302
  logging_steps=25,
303
  logging_first_step=True,
304
  load_best_model_at_end=True,
 
295
  warmup_ratio=0.1,
296
  bf16=True,
297
  eval_strategy="steps",
298
+ eval_steps=25,
299
  save_strategy="steps",
300
+ save_steps=25,
301
+ save_total_limit=5,
302
  logging_steps=25,
303
  logging_first_step=True,
304
  load_best_model_at_end=True,