Skip to content

Commit ec125e7

Browse files
committed
formatting changes
1 parent 3e0fa4c commit ec125e7

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

nncf/common/graph/utils.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -157,11 +157,13 @@ def get_weight_shape_legacy(layer_attributes: WeightedLayerAttributes) -> List[i
157157
return [
158158
layer_attributes.out_channels,
159159
layer_attributes.in_channels // layer_attributes.groups,
160-
*layer_attributes.kernel_size]
160+
*layer_attributes.kernel_size
161+
]
161162
return [
162163
layer_attributes.in_channels,
163164
layer_attributes.out_channels // layer_attributes.groups,
164-
*layer_attributes.kernel_size]
165+
*layer_attributes.kernel_size
166+
]
165167

166168
if isinstance(layer_attributes, GroupNormLayerAttributes):
167169
return [layer_attributes.num_channels]
@@ -204,4 +206,4 @@ def get_num_filters_legacy(layer_attributes: WeightedLayerAttributes) -> int:
204206
"""
205207
weight_shape = layer_attributes.get_weight_shape_legacy()
206208
return weight_shape[layer_attributes.get_target_dim_for_compression_legacy()]
207-
209+

0 commit comments

Comments
 (0)