louis_lifu
/
AReaL
mirror of https://github.com/inclusionAI/AReaL
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

PullRequest: 321 fix checkpoint save dir 

Merge branch xss/debug of git@code.alipay.com:inclusionAI/AReaL.git into gh
https://code.alipay.com/inclusionAI/AReaL/pull_requests/321

Reviewed-by: 博惟 <bowei.fw@antgroup.com>


* fix checkpoint save dir
This commit is contained in:
温差 2025-07-03 10:03:33 +08:00 committed by 博惟
parent 623f7c7407
commit ed6bcd66a7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
realhf/system/model_function_call.py
View File

@ -216,7 +216,7 @@ class ModelFunctionCall:
for p in payloads.values():
p.post_hooks.append("save")
save_dir = os.path.join(
constants.get_log_path(self.args),
constants.get_save_path(self.args),
rpc.model_name.role,
f"epoch{ctrl.step_info.epoch + 1}"
f"epochstep{ctrl.step_info.epoch_step + 1}"