NeMo SSL collection API#
Model Classes#
Mixins#
- class nemo.core.classes.mixins.access_mixins.AccessMixin[source]#
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)[source]#
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.