nemo_rl.weight_sync.checkpoint_engine_config

Module Contents

Functions

checkpoint_engine_refit_config

Translate a checkpoint-engine refit scope into the internal schema.

API

nemo_rl.weight_sync.checkpoint_engine_config.checkpoint_engine_refit_config(

generation_config: nemo_rl.models.generation.interfaces.GenerationConfig,

) → nemo_rl.models.generation.interfaces.CheckpointEngineConfig | None

Translate a checkpoint-engine refit scope into the internal schema.