diff --git a/pretrainedmodels/models/resnext.py b/pretrainedmodels/models/resnext.py index 394b8182..2ea086d8 100644 --- a/pretrainedmodels/models/resnext.py +++ b/pretrainedmodels/models/resnext.py @@ -39,7 +39,7 @@ class ResNeXt101_32x4d(nn.Module): def __init__(self, num_classes=1000): super(ResNeXt101_32x4d, self).__init__() self.num_classes = num_classes - self.features = resnext101_32x4d_features + self.features = resnext101_32x4d_features() self.avg_pool = nn.AvgPool2d((7, 7), (1, 1)) self.last_linear = nn.Linear(2048, num_classes) @@ -60,7 +60,7 @@ class ResNeXt101_64x4d(nn.Module): def __init__(self, num_classes=1000): super(ResNeXt101_64x4d, self).__init__() self.num_classes = num_classes - self.features = resnext101_64x4d_features + self.features = resnext101_64x4d_features() self.avg_pool = nn.AvgPool2d((7, 7), (1, 1)) self.last_linear = nn.Linear(2048, num_classes) diff --git a/pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py b/pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py index 21ba0b25..1fdf29c0 100644 --- a/pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py +++ b/pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py @@ -42,7 +42,7 @@ def identity(x): return x def add(x, y): return x + y -resnext101_32x4d_features = nn.Sequential( # Sequential, +resnext101_32x4d_features = lambda: nn.Sequential( # Sequential, nn.Conv2d(3,64,(7, 7),(2, 2),(3, 3),1,1,bias=False), nn.BatchNorm2d(64), nn.ReLU(), diff --git a/pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py b/pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py index ac1d6bcc..9ccb065c 100644 --- a/pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py +++ b/pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py @@ -42,7 +42,7 @@ def identity(x): return x def add(x, y): return x + y -resnext101_64x4d_features = nn.Sequential(#Sequential, +resnext101_64x4d_features = lambda: nn.Sequential(#Sequential, nn.Conv2d(3, 64, (7, 7), (2, 2), (3, 3), 1, 1, bias = False), nn.BatchNorm2d(64), nn.ReLU(), diff --git a/pretrainedmodels/version.py b/pretrainedmodels/version.py index a9df0271..ce5153af 100644 --- a/pretrainedmodels/version.py +++ b/pretrainedmodels/version.py @@ -1,2 +1,2 @@ from __future__ import print_function, division, absolute_import -__version__ = '0.7.4' +__version__ = '0.7.5'