#! /usr/bin/python # -*- encoding: utf-8 -*- import torchvision def MainModel(nOut=256, **kwargs): return torchvision.models.resnext50_32x4d(num_classes=nOut)