av-codes commited on
Commit
9760e4f
·
verified ·
1 Parent(s): da6eea2

fix: match eval and save strategy (both steps/4000)

Browse files
Files changed (1) hide show
  1. 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="epoch",
 
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,