提交 e204f230 编写于 作者: H HexToString

fix bug

上级 b306084f
...@@ -35,7 +35,7 @@ def single_func(idx, resource): ...@@ -35,7 +35,7 @@ def single_func(idx, resource):
batch_size=1) batch_size=1)
start = time.time() start = time.time()
for data in train_reader(): 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() end = time.time()
return [[end - start]] return [[end - start]]
elif args.request == "http": elif args.request == "http":
......
...@@ -155,7 +155,7 @@ class Client(object): ...@@ -155,7 +155,7 @@ class Client(object):
file_path_list = [] file_path_list = []
for single_model_config in model_config_path_list: for single_model_config in model_config_path_list:
if os.path.isdir(single_model_config): 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)) single_model_config))
elif os.path.isfile(single_model_config): elif os.path.isfile(single_model_config):
file_path_list.append(single_model_config) file_path_list.append(single_model_config)
...@@ -574,7 +574,7 @@ class MultiLangClient(object): ...@@ -574,7 +574,7 @@ class MultiLangClient(object):
file_path_list = [] file_path_list = []
for single_model_config in model_config_path_list: for single_model_config in model_config_path_list:
if os.path.isdir(single_model_config): 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)) single_model_config))
elif os.path.isfile(single_model_config): elif os.path.isfile(single_model_config):
file_path_list.append(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.
先完成此消息的编辑!
想要评论请 注册