<reponame>joonson/face_trainer
#! /usr/bin/python
# -*- encoding: utf-8 -*-

import torchvision

def MainModel(nOut=256, **kwargs):
    
    return torchvision.models.resnet18(num_classes=nOut)

#! /usr/bin/python
# -*- encoding: utf-8 -*-

import torchvision

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