提交 587e1906 编写于 作者: Q qiaolongfei

use logdir to find model file

上级 5f81259a
...@@ -175,7 +175,7 @@ def graph(): ...@@ -175,7 +175,7 @@ def graph():
# run = request.args.get('run') # run = request.args.get('run')
# model_json_str = mock_data.graph_data() # model_json_str = mock_data.graph_data()
# model_json = json.loads(model_json_str) # model_json = json.loads(model_json_str)
model_json = vdl_graph.load_model(server_path + "/mock/squeezenet_model.pb") model_json = vdl_graph.load_model(options.logdir + "/model.pb")
result = gen_result(0, "", model_json) result = gen_result(0, "", model_json)
return Response(json.dumps(result), mimetype='application/json') return Response(json.dumps(result), mimetype='application/json')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册