From fa4265a7bc1b2038bd87b3d8fd79d3c6ecb11d89 Mon Sep 17 00:00:00 2001 From: Chenhe Gu Date: Mon, 13 Jul 2026 17:00:07 +0800 Subject: [PATCH] fix --- slime/utils/arguments.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/slime/utils/arguments.py b/slime/utils/arguments.py index 0a79b99ded..be64d3bb3d 100644 --- a/slime/utils/arguments.py +++ b/slime/utils/arguments.py @@ -1623,6 +1623,20 @@ def _apply_megatron_role_overrides(base_args, overrides, role): if "disable_param_buffers_cpu_backup" not in overrides: role_args.disable_param_buffers_cpu_backup = False + # If the critic's own load has no checkpoint yet, fall back to the + # actor's resolved load. + critic_load = role_args.load + if ( + critic_load is None + or not os.path.exists(critic_load) + or not os.path.exists(os.path.join(critic_load, "latest_checkpointed_iteration.txt")) + ): + role_args.load = base_args.load + role_args.no_load_optim = True + role_args.no_load_rng = True + role_args.finetune = True + role_args.start_rollout_id = 0 + return role_args