Parcourir la source

Zero raw_data only if exists

Lutz Roeder il y a 8 ans
Parent
commit
8b1946f368
1 fichiers modifiés avec 2 ajouts et 1 suppressions
  1. 2 1
      src/netron.py

+ 2 - 1
src/netron.py

@@ -101,7 +101,8 @@ def optimize_onnx(model):
         del tensor.int32_data[:]
         del tensor.int64_data[:]
         del tensor.float_data[:]
-        tensor.raw_data = b''
+        if tensor.HasField('raw_data'):
+            tensor.raw_data = b''
     # Remove raw initializer data
     onnx_model = ModelProto()
     try: