bspysmg.data.dataset#

File containing a class for loading sampling data as a dataset, as well as a function for loading the dataset into a PyTorch dataloader.

Functions

get_dataloaders(configs)

Loads all the datasets specified in the dataset_paths list key of the configurations dictionary and creates a dataloader.

get_info_dict(training_configs, sampling_configs)

Retrieve the info dictionary given the training configs and the sampling configs.

Classes

ModelDataset(filename[, steps])