hy2dl.datasetzoo

hy2dl.datasetzoo.get_dataset(cfg: Config) type[BaseDataset]

Get data set class, depending on the run configuration.

Parameters:

cfg (Config) – Configuration file.

Returns:

The dataset class corresponding to the configuration.

Return type:

BaseDataset