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]
class labml.utils.data.pytorch.CsvDataset(file_path: str, y_cols: List, x_cols: List, train: bool = True, transform: Callable = <function CsvDataset.<lambda>>, test_fraction: float = 0.0, nrows: int = None)[source]