Ver Fonte

Fix python.py to strip ONNX raw_data

Lutz Roeder há 8 anos atrás
pai
commit
248ff92f67
2 ficheiros alterados com 2 adições e 2 exclusões
  1. 1 1
      package.json
  2. 1 1
      src/netron.py

+ 1 - 1
package.json

@@ -17,8 +17,8 @@
         "postinstall": "electron-builder install-app-deps"
     },
     "dependencies": {
-        "dagre": "x.x.x",
         "d3": "x.x.x",
+        "dagre": "x.x.x",
         "electron-updater": "^2.17.6",
         "flatbuffers": "x.x.x",
         "handlebars": "x.x.x",

+ 1 - 1
src/netron.py

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