ReferenceFull Library ReferenceNemo AutomodelNemo AutomodelShared

nemo_automodel.shared.parameter_names

View as Markdown

Utilities for stable logical parameter names.

Module Contents

Functions

NameDescription
canonical_parameter_fqnRemove activation-checkpoint wrapper components from a parameter FQN.

API

nemo_automodel.shared.parameter_names.canonical_parameter_fqn(
    name: str
) -> str

Remove activation-checkpoint wrapper components from a parameter FQN.

PyTorch exposes CheckpointWrapper as a registered child module through named_parameters(), but strips the same component from state-dict keys. Canonicalizing name-sensitive consumers keeps their logical names aligned with the stable state-dict contract.

Parameters:

name
str

Module-qualified parameter name.

Returns: str

The logical parameter name without checkpoint-wrapper components.