@@ -138,22 +138,24 @@ def forward(self, x):
138
138
class QuantizableInceptionE (inception_module .InceptionE ):
139
139
def __init__ (self , * args , ** kwargs ):
140
140
super (QuantizableInceptionE , self ).__init__ (conv_block = QuantizableBasicConv2d , * args , ** kwargs )
141
- self .myop = nn .quantized .FloatFunctional ()
141
+ self .myop1 = nn .quantized .FloatFunctional ()
142
+ self .myop2 = nn .quantized .FloatFunctional ()
143
+ self .myop3 = nn .quantized .FloatFunctional ()
142
144
143
145
def _forward (self , x ):
144
146
branch1x1 = self .branch1x1 (x )
145
147
146
148
branch3x3 = self .branch3x3_1 (x )
147
149
branch3x3 = [self .branch3x3_2a (branch3x3 ), self .branch3x3_2b (branch3x3 )]
148
- branch3x3 = self .myop .cat (branch3x3 , 1 )
150
+ branch3x3 = self .myop1 .cat (branch3x3 , 1 )
149
151
150
152
branch3x3dbl = self .branch3x3dbl_1 (x )
151
153
branch3x3dbl = self .branch3x3dbl_2 (branch3x3dbl )
152
154
branch3x3dbl = [
153
155
self .branch3x3dbl_3a (branch3x3dbl ),
154
156
self .branch3x3dbl_3b (branch3x3dbl ),
155
157
]
156
- branch3x3dbl = self .myop .cat (branch3x3dbl , 1 )
158
+ branch3x3dbl = self .myop2 .cat (branch3x3dbl , 1 )
157
159
158
160
branch_pool = F .avg_pool2d (x , kernel_size = 3 , stride = 1 , padding = 1 )
159
161
branch_pool = self .branch_pool (branch_pool )
@@ -163,7 +165,7 @@ def _forward(self, x):
163
165
164
166
def forward (self , x ):
165
167
outputs = self ._forward (x )
166
- return self .myop .cat (outputs , 1 )
168
+ return self .myop3 .cat (outputs , 1 )
167
169
168
170
169
171
class QuantizableInceptionAux (inception_module .InceptionAux ):
0 commit comments