Ver código fonte

Zero raw_data only if exists

Lutz Roeder 8 anos atrás
pai
commit
8b1946f368
1 arquivos alterados com 2 adições e 1 exclusões
  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: