可视化graph的时候报错KeyError: 'dim'
Created by: Michael3444
Traceback (most recent call last): File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 2292, in wsgi_app response = self.full_dispatch_request() File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1815, in full_dispatch_request rv = self.handle_user_exception(e) File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1718, in handle_user_exception reraise(exc_type, exc_value, tb) File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1813, in full_dispatch_request rv = self.dispatch_request() File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1799, in dispatch_request return self.view_functionsrule.endpoint File "/usr/local/bin/visualdl", line 324, in graph json_str = vdl_graph.draw_onnx_graph(args.model_pb) File "/usr/local/lib/python2.7/dist-packages/visualdl/server/graph.py", line 491, in draw_onnx_graph json_str = load_model(model_pb_path) File "/usr/local/lib/python2.7/dist-packages/visualdl/server/graph.py", line 369, in load_model model_json = to_IR_json(model_pb_path) File "/usr/local/lib/python2.7/dist-packages/visualdl/server/graph.py", line 363, in to_IR_json reorganize_inout(model_json, 'input') File "/usr/local/lib/python2.7/dist-packages/visualdl/server/graph.py", line 51, in reorganize_inout shape = [dim['dimValue'] for dim in tensor_type['shape']['dim']] KeyError: 'dim'