diff --git a/verl/workers/fsdp_workers.py b/verl/workers/fsdp_workers.py index 0b641f9c..8e5eebc0 100644 --- a/verl/workers/fsdp_workers.py +++ b/verl/workers/fsdp_workers.py @@ -435,7 +435,7 @@ def load_checkpoint(self, path: str, remove_ckpt_after_load: bool = False): offload_fsdp_model(self.fsdp_module) if self._use_optimizer_offload: - offload_fsdp_optimizer(self.actor_optimizer) + offload_fsdp_optimizer(self.optimizer) """ActorRolloutRefWorker"""