CerebNet.data_loader.loader¶

CerebNet.data_loader.loader.get_dataloader(cfg, mode)[source]¶

Create the dataset and pytorch data loader.

Parameters:

cfgobject: Configuration object containing data loading parameters.
modestr: Loading mode - either ‘train’ or ‘val’.

Returns:

DataLoader: PyTorch DataLoader configured based on the mode.

Raises:

ValueError: If mode is not ‘train’ or ‘val’.