src/models/__init__.py

from src.models.dca_net import DCANet
from src.models.baselines import ResNet3D18, ResNet2D18SliceLevel, get_baseline_model

__all__ = ['DCANet', 'ResNet3D18', 'ResNet2D18SliceLevel', 'get_baseline_model']