From 4d5788263a98a79f9f644bb2c55baf815e22c85b Mon Sep 17 00:00:00 2001 From: Sasank Chilamkurthy Date: Wed, 6 May 2020 18:58:57 +0530 Subject: [PATCH 1/2] add hubconf --- hubconf.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 hubconf.py diff --git a/hubconf.py b/hubconf.py new file mode 100644 index 00000000..2a041056 --- /dev/null +++ b/hubconf.py @@ -0,0 +1,47 @@ +dependencies = ['torch', 'torchvision', 'munch', 'tqdm', 'scipy'] + +from pretrainedmodels.models.fbresnet import fbresnet152 +from pretrainedmodels.models.cafferesnet import cafferesnet101 +from pretrainedmodels.models.bninception import bninception +from pretrainedmodels.models.resnext import resnext101_32x4d +from pretrainedmodels.models.resnext import resnext101_64x4d +from pretrainedmodels.models.inceptionv4 import inceptionv4 +from pretrainedmodels.models.inceptionresnetv2 import inceptionresnetv2 +from pretrainedmodels.models.nasnet import nasnetalarge +from pretrainedmodels.models.nasnet_mobile import nasnetamobile +from pretrainedmodels.models.torchvision_models import alexnet +from pretrainedmodels.models.torchvision_models import densenet121 +from pretrainedmodels.models.torchvision_models import densenet169 +from pretrainedmodels.models.torchvision_models import densenet201 +from pretrainedmodels.models.torchvision_models import densenet161 +from pretrainedmodels.models.torchvision_models import resnet18 +from pretrainedmodels.models.torchvision_models import resnet34 +from pretrainedmodels.models.torchvision_models import resnet50 +from pretrainedmodels.models.torchvision_models import resnet101 +from pretrainedmodels.models.torchvision_models import resnet152 +from pretrainedmodels.models.torchvision_models import inceptionv3 +from pretrainedmodels.models.torchvision_models import squeezenet1_0 +from pretrainedmodels.models.torchvision_models import squeezenet1_1 +from pretrainedmodels.models.torchvision_models import vgg11 +from pretrainedmodels.models.torchvision_models import vgg11_bn +from pretrainedmodels.models.torchvision_models import vgg13 +from pretrainedmodels.models.torchvision_models import vgg13_bn +from pretrainedmodels.models.torchvision_models import vgg16 +from pretrainedmodels.models.torchvision_models import vgg16_bn +from pretrainedmodels.models.torchvision_models import vgg19_bn +from pretrainedmodels.models.torchvision_models import vgg19 +from pretrainedmodels.models.dpn import dpn68 +from pretrainedmodels.models.dpn import dpn68b +from pretrainedmodels.models.dpn import dpn92 +from pretrainedmodels.models.dpn import dpn98 +from pretrainedmodels.models.dpn import dpn131 +from pretrainedmodels.models.dpn import dpn107 +from pretrainedmodels.models.xception import xception +from pretrainedmodels.models.senet import senet154 +from pretrainedmodels.models.senet import se_resnet50 +from pretrainedmodels.models.senet import se_resnet101 +from pretrainedmodels.models.senet import se_resnet152 +from pretrainedmodels.models.senet import se_resnext50_32x4d +from pretrainedmodels.models.senet import se_resnext101_32x4d +from pretrainedmodels.models.pnasnet import pnasnet5large +from pretrainedmodels.models.polynet import polynet From fcc7564ce53339ab91ef8fe5b0988d3e634c72be Mon Sep 17 00:00:00 2001 From: Sasank Chilamkurthy Date: Wed, 6 May 2020 19:13:11 +0530 Subject: [PATCH 2/2] scipy not neccessary --- hubconf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index 2a041056..e1a94daa 100644 --- a/hubconf.py +++ b/hubconf.py @@ -1,4 +1,4 @@ -dependencies = ['torch', 'torchvision', 'munch', 'tqdm', 'scipy'] +dependencies = ['torch', 'torchvision', 'munch', 'tqdm'] from pretrainedmodels.models.fbresnet import fbresnet152 from pretrainedmodels.models.cafferesnet import cafferesnet101