ModuleNotFoundError: No module named 'capslayer.data.datasets.stanford_drone'

[akorez]

Hi, I created the datareader and datawriter according to my own dataset by following the tutorials. I made the necessary changes to config.py and main.py files. But I get an error as mentioned in the title. How can I create my own module similar to the cifar10 module? Please, help me! Thanks...

[naturomics]

python can not find your dataloader, you should import your module to this init.py file

[akorez]

I added my own module into the init.py file. Here's the code. I still get the same error.

from future import absolute_import from future import division from future import print_function

from . import mnist from . import fashion_mnist from . import cifar10 from . import cifar100 from . import stanford_drone

all = ['mnist', 'fashion_mnist', 'cifar10', 'cifar100','stanford_drone']