Skip to content

Commit def4658

Browse files
committed
Merge: [ResNet/PyT] Fix Resnet BasicBlock constructor
2 parents 2586ee3 + 0c4310b commit def4658

File tree

1 file changed

+4
-2
lines changed
  • PyTorch/Classification/ConvNets/image_classification/models

1 file changed

+4
-2
lines changed

PyTorch/Classification/ConvNets/image_classification/models/resnet.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,14 +63,16 @@ def __init__(
6363
stride=1,
6464
cardinality=1,
6565
downsample=None,
66+
fused_se=True,
6667
last_bn_0_init=False,
68+
trt=False,
6769
):
6870
super(BasicBlock, self).__init__()
69-
self.conv1 = builder.conv3x3(inplanes, planes, stride, cardinality=cardinality)
71+
self.conv1 = builder.conv3x3(inplanes, planes, stride, groups=cardinality)
7072
self.bn1 = builder.batchnorm(planes)
7173
self.relu = builder.activation()
7274
self.conv2 = builder.conv3x3(
73-
planes, planes * expansion, cardinality=cardinality
75+
planes, planes * expansion, groups=cardinality
7476
)
7577
self.bn2 = builder.batchnorm(planes * expansion, zero_init=last_bn_0_init)
7678
self.downsample = downsample

0 commit comments

Comments
 (0)