ResNeXt50.py 170 B

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