nemo_rl.environments.checkpointing#
Pick the environment backend’s rollout checkpoint for a run.
Submodules#
Package Contents#
Functions#
Return the environment checkpoint this run’s config asks for. |
API#
- nemo_rl.environments.checkpointing.build_environment_checkpoint(
- master_config: nemo_rl.algorithms.single_controller_utils.config.MasterConfig,
- *,
- env_handles: dict[str, Any],
- rollout_manager: nemo_rl.experience.rollout_manager.RolloutManager,
- dp_client: Any,
Return the environment checkpoint this run’s config asks for.
Only turn-level recovery keeps environment state, and setup already requires NeMo Gym for it. A new backend that keeps rollout state adds its branch here.