瀏覽代碼

Zero raw_data only if exists

Lutz Roeder 8 年之前
父節點
當前提交
8b1946f368
共有 1 個文件被更改,包括 2 次插入1 次删除
  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: