fix resnext bug

Author: Tete Xiao

models.py

@@ -2,6 +2,7 @@
 import torch.nn as nn
 import torchvision
 import resnet
+import resnext
 from lib.nn import SynchronizedBatchNorm2d
 
 
@@ -109,6 +110,9 @@ def build_encoder(self, arch='resnet50_dilated8', fc_dim=512, weights=''):
             orig_resnet = resnet.__dict__['resnet101'](pretrained=pretrained)
             net_encoder = ResnetDilated(orig_resnet,
                                         dilate_scale=16)
+        elif arch == 'resnext101':
+            orig_resnext = resnext.__dict__['resnext101'](pretrained=pretrained)
+            net_encoder = Resnet(orig_resnext) # we can still use class Resnet
         else:
             raise Exception('Architecture undefined!')
 

resnext.py

@@ -15,7 +15,7 @@
 
 
 model_urls = {
-    #'resnet50': 'http://sceneparsing.csail.mit.edu/model/pretrained_resnet/resnet50-imagenet.pth',
+    #'resnext50': 'http://sceneparsing.csail.mit.edu/model/pretrained_resnet/resnext50-imagenet.pth',
     'resnext101': 'http://sceneparsing.csail.mit.edu/model/pretrained_resnet/resnext101-imagenet.pth'
 }