We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 04a05b5 commit 2664489Copy full SHA for 2664489
src/nncf/experimental/common/tensor_statistics/statistics.py
@@ -110,8 +110,8 @@ class MeanTensorStatistic(TensorStatistic):
110
shape: tuple[int, ...]
111
112
def __post_init__(self):
113
- if isinstance(self.shape, (deque, list)):
114
- # If NoopAggregator was used, shape can be a sequence containing a single tensor
+ if isinstance(self.shape[0], Tensor):
+ # If in-place shape reducer and Noop aggregator were used, shape is a sequence containing a single tensor
115
self.shape = tuple(self.shape[0].data.tolist())
116
117
def __eq__(self, other: TensorStatistic):
0 commit comments