This PR contains the structure for a BaseDataset that inherits from torch.utils.data.Dataset and contains functions to retrieve, manipulate and cache the downloaded dataset.
It also contains an implementation of a QM9Dataset that provides functions that control from which remote location the dataset is retrieved and the transformations to generate an opimized npy file. It also implements the __getitem__ and __len__ calls necessary for the torch.utils.data.Dataset class.
Description
Outline the basic dataset structure
This PR contains the structure for a
BaseDatasetthat inherits fromtorch.utils.data.Datasetand contains functions to retrieve, manipulate and cache the downloaded dataset. It also contains an implementation of aQM9Datasetthat provides functions that control from which remote location the dataset is retrieved and the transformations to generate an opimized npy file. It also implements the__getitem__and__len__calls necessary for thetorch.utils.data.Datasetclass.Todos
Status