提交 6d9d2e1e 编写于 作者: W wangjiawei04

pre-commit code test=serving

上级 0cbffcc7
FILE(GLOB common_srcs ${CMAKE_CURRENT_LIST_DIR}/constant.cpp) FILE(GLOB common_srcs ${CMAKE_CURRENT_LIST_DIR}/constant.cpp)
LIST(APPEND pdserving_srcs ${common_srcs}) LIST(APPEND pdserving_srcs ${common_srcs})
...@@ -28,11 +28,8 @@ batch = 1 ...@@ -28,11 +28,8 @@ batch = 1
buf_size = 100 buf_size = 100
dataset = criteo.CriteoDataset() dataset = criteo.CriteoDataset()
dataset.setup(1000001) dataset.setup(1000001)
test_filelists = [ test_filelists = ["{}/part-test".format(sys.argv[2])]
"{}/part-test".format(sys.argv[2]) reader = dataset.infer_reader(test_filelists, batch, buf_size)
]
reader = dataset.infer_reader(test_filelists, batch,
buf_size)
label_list = [] label_list = []
prob_list = [] prob_list = []
start = time.time() start = time.time()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册