fix typo

Signed-off-by: yuwenzho <[email protected]>
intel · Sep 20, 2023 · 0bde90c · 0bde90c
1 parent 12dbab2
commit 0bde90c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neural_compressor/model/onnx_model.py b/neural_compressor/model/onnx_model.py
@@ -65,10 +65,10 @@ def check_large_model(self):
         """Check model > 2GB."""
         init_size = 0
         for init in self._model.graph.initializer:
-            # if size of the initializer > 2GB, return True
+            # if raise error of initializer size > 2GB, return True
             try:
                 init_bytes = init.SerializeToString()
-                init_size += sys.getsizeof(init.SerializeToString())
+                init_size += sys.getsizeof(init_bytes)
             except Exception as e:
                 if "exceeds maximum protobuf size of 2GB" in str(e):
                     return True