from . import classification, segmentation

from torchvision import models as torchvision_models