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.

classmethod is_access_enabled()

register_accessible_tensor(name, tensor)

Register tensor for later use.

reset_registry(registry_key: Optional[str] = None)

Reset the registries of all named sub-modules

classmethod set_access_enabled(access_enabled: bool)

classmethod update_access_cfg(cfg: dict)