CerebNet.data_loader.loader¶

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

Creating the dataset and pytorch data loader

Args:: cfg: mode: loading data for train, val and test mode
Returns:: the Dataloader