Update aggregator.py

openvinotoolkit · Oct 8, 2024 · 37ce7a6 · 37ce7a6
1 parent 7fdb664
commit 37ce7a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nncf/common/tensor_statistics/aggregator.py b/nncf/common/tensor_statistics/aggregator.py
@@ -87,7 +87,7 @@ def collect_statistics(self, model: TModel, graph: NNCFGraph) -> None:
         if processed_samples == 0:
             raise nncf.ValidationError(EMPTY_DATASET_ERROR)
 
-        if self.stat_subset_size > processed_samples:
+        if self.stat_subset_size is not None and self.stat_subset_size > processed_samples:
             nncf_logger.warning(
                 f"Dataset contains only {processed_samples} samples, "
                 f"smaller than the requested subset size {self.stat_subset_size}."