Eval and save every 25 steps for detailed curves
Browse files- 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=
|
| 299 |
save_strategy="steps",
|
| 300 |
-
save_steps=
|
| 301 |
-
save_total_limit=
|
| 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,
|