Create different PyTorch DataModule and Dataset options

[alawryaguila]

Currently there is only one datamodule option where the data is inputted and saved and called by the dataset module when required. Another option would be to input the data path to the datamodule and the dataset module loads the data when needed.

[alawryaguila]

Requires editing fit function to remove checking of input data shape. Input data might be list of indices to load data rather than actual data.