fix: match eval and save strategy (both steps/4000)
Browse files- train_hrm_text_pi.py +2 -1
train_hrm_text_pi.py
CHANGED
|
@@ -928,7 +928,8 @@ def main():
|
|
| 928 |
weight_decay=0.01,
|
| 929 |
warmup_steps=500 if not args.test else 0,
|
| 930 |
lr_scheduler_type="cosine",
|
| 931 |
-
eval_strategy="
|
|
|
|
| 932 |
save_strategy="steps",
|
| 933 |
save_steps=4000,
|
| 934 |
load_best_model_at_end=True,
|
|
|
|
| 928 |
weight_decay=0.01,
|
| 929 |
warmup_steps=500 if not args.test else 0,
|
| 930 |
lr_scheduler_type="cosine",
|
| 931 |
+
eval_strategy="steps",
|
| 932 |
+
eval_steps=4000,
|
| 933 |
save_strategy="steps",
|
| 934 |
save_steps=4000,
|
| 935 |
load_best_model_at_end=True,
|