NeMo SSL collection API#

Model Classes#

Mixins#

class nemo.core.classes.mixins.access_mixins.AccessMixin

Bases: ABC

Allows access to output of intermediate layers of a model

property access_cfg

Returns: The global access config shared across all access mixin modules.

classmethod get_module_registry(module: torch.nn.Module)

Extract all registries from named submodules, return dictionary where the keys are the flattened module names, the values are the internal registry of each such module.

register_accessible_tensor(name, tensor)

Register tensor for later use.

reset_registry(registry_key: str | None = None)

Reset the registries of all named sub-modules