Error by MemoryDataset 2

[erogol]

I try to use my custom dataset with the given AutoEncoder example but when it starts it raises following error

aining)
    225                      str(type(self.model)), func_i+1, len(self.train_functions), self.n_epoch, str(continue_training))
    226 
--> 227             log.debug("Train dataset size is: %s", self.dataset.getDataShape(datasets.TRAIN))
    228             if self.dataset.hasSubset(datasets.VALID):
    229                 log.debug("Valid dataset size is: %s", self.dataset.getDataShape(datasets.VALID))

/home/retina18/Downloads/opendeep/opendeep/data/dataset.py in getDataShape(self, subset)
    111         if subset is TRAIN:
    112             log.error("No training shape implemented")
--> 113             raise NotImplementedError("No training shape implemented")
    114         elif subset is VALID:
    115             log.error("No valid shape implemented")

NotImplementedError: No training shape implemented

[vitruvianscience]

Should be fixed now, thanks for pointing it out!