models/__init__.py

from .models import ArcMarginModel
from .models import ResNet
from .models import IRBlock
from .models import SEBlock