Utils

labml.utils.pytorch.store_model_indicators(model: torch.nn.modules.module.Module, *, model_name: str = 'model')[source]
labml.utils.pytorch.get_modules(configs: labml.configs.BaseConfigs)[source]