提交 e204f230 编写于 作者: H HexToString

fix bug

上级 b306084f
......@@ -35,7 +35,7 @@ def single_func(idx, resource):
batch_size=1)
start = time.time()
for data in train_reader():
fetch_map = client.predict(feed={"x": data[0][0]}, fetch=["price"])
fetch_map = client.predict(feed={"x": data[0][0]}, fetch=["price"],batch=True)
end = time.time()
return [[end - start]]
elif args.request == "http":
......
......@@ -155,7 +155,7 @@ class Client(object):
file_path_list = []
for single_model_config in model_config_path_list:
if os.path.isdir(single_model_config):
file_path_list.append("{}/serving_server_conf.prototxt".format(
file_path_list.append("{}/serving_client_conf.prototxt".format(
single_model_config))
elif os.path.isfile(single_model_config):
file_path_list.append(single_model_config)
......@@ -574,7 +574,7 @@ class MultiLangClient(object):
file_path_list = []
for single_model_config in model_config_path_list:
if os.path.isdir(single_model_config):
file_path_list.append("{}/serving_server_conf.prototxt".format(
file_path_list.append("{}/serving_client_conf.prototxt".format(
single_model_config))
elif os.path.isfile(single_model_config):
file_path_list.append(single_model_config)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册