[PaddlePaddle Hackathon] add ResNeXt

python/paddle/tests/test_pretrained_model.py

@@ -53,7 +53,8 @@ def infer(self, arch):
 
     def test_models(self):
         arches = [
-            'mobilenet_v1', 'mobilenet_v2', 'resnet18', 'vgg16', 'alexnet'
+            'mobilenet_v1', 'mobilenet_v2', 'resnet18', 'vgg16', 'alexnet',
+            'resnext50_32x4d'
         ]
         for arch in arches:
             self.infer(arch)

python/paddle/tests/test_vision_models.py

@@ -73,6 +73,24 @@ def test_resnet152(self):
     def test_alexnet(self):
         self.models_infer('alexnet')
 
+    def test_resnext50_32x4d(self):
+        self.models_infer('resnext50_32x4d')
+
+    def test_resnext50_64x4d(self):
+        self.models_infer('resnext50_64x4d')
+
+    def test_resnext101_32x4d(self):
+        self.models_infer('resnext101_32x4d')
+
+    def test_resnext101_64x4d(self):
+        self.models_infer('resnext101_64x4d')
+
+    def test_resnext152_32x4d(self):
+        self.models_infer('resnext152_32x4d')
+
+    def test_resnext152_64x4d(self):
+        self.models_infer('resnext152_64x4d')
+
     def test_vgg16_num_classes(self):
         vgg16 = models.__dict__['vgg16'](pretrained=False, num_classes=10)
 

python/paddle/vision/__init__.py

@@ -46,6 +46,13 @@
 from .models import LeNet  # noqa: F401
 from .models import AlexNet  # noqa: F401
 from .models import alexnet  # noqa: F401
+from .models import ResNeXt  # noqa: F401
+from .models import resnext50_32x4d  # noqa: F401