File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
PyTorch/Classification/ConvNets/image_classification/models Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments