models/__init__.py

from models.CNN import CNN as CNN
from models.ResNet18_1d import ResNet1D as ResNet