code/datamodules/__init__.py

from .base import ImageDataModule
from .image_classification import CIFAR10DataModule, MNISTDataModule
from .visual_qa import CIFAR10QADataModule, ToyQADataModule