We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 93dc6f5 commit fdbffe7Copy full SHA for fdbffe7
recipe/one_step_off_policy/fsdp_workers.py
@@ -220,6 +220,7 @@ def init_model(self):
220
221
rollout_config: RolloutConfig = omega_conf_to_dataclass(self.config.rollout)
222
model_config: HFModelConfig = omega_conf_to_dataclass(self.config.model, dataclass_type=HFModelConfig)
223
+ self.model_config = model_config
224
225
log_gpu_memory_usage(f"Before building {rollout_name} rollout", logger=logger)
226
rollout = get_rollout_class(rollout_config.name, rollout_config.mode)(
0 commit comments