ResNet18.py 163 B

12345678
  1. #! /usr/bin/python
  2. # -*- encoding: utf-8 -*-
  3. import torchvision
  4. def MainModel(nOut=256, **kwargs):
  5. return torchvision.models.resnet18(num_classes=nOut)