File tree 1 file changed +5
-3
lines changed
1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -157,11 +157,13 @@ def get_weight_shape_legacy(layer_attributes: WeightedLayerAttributes) -> List[i
157
157
return [
158
158
layer_attributes .out_channels ,
159
159
layer_attributes .in_channels // layer_attributes .groups ,
160
- * layer_attributes .kernel_size ]
160
+ * layer_attributes .kernel_size
161
+ ]
161
162
return [
162
163
layer_attributes .in_channels ,
163
164
layer_attributes .out_channels // layer_attributes .groups ,
164
- * layer_attributes .kernel_size ]
165
+ * layer_attributes .kernel_size
166
+ ]
165
167
166
168
if isinstance (layer_attributes , GroupNormLayerAttributes ):
167
169
return [layer_attributes .num_channels ]
@@ -204,4 +206,4 @@ def get_num_filters_legacy(layer_attributes: WeightedLayerAttributes) -> int:
204
206
"""
205
207
weight_shape = layer_attributes .get_weight_shape_legacy ()
206
208
return weight_shape [layer_attributes .get_target_dim_for_compression_legacy ()]
207
-
209
+
You can’t perform that action at this time.
0 commit comments