提交 8ef4b9c1 编写于 作者: M MRXLT

fix criteo_ctr_with_cube/benchmark_batch.py

上级 6eddb7a8
...@@ -56,7 +56,7 @@ def single_func(idx, resource): ...@@ -56,7 +56,7 @@ def single_func(idx, resource):
feed_dict["embedding_{}.tmp_0".format(i - 1)] = data[0][ feed_dict["embedding_{}.tmp_0".format(i - 1)] = data[0][
i] i]
feed_batch.append(feed_dict) feed_batch.append(feed_dict)
result = client.batch_predict(feed=feed_batch, fetch=fetch) result = client.predict(feed=feed_batch, fetch=fetch)
else: else:
print("unsupport batch size {}".format(args.batch_size)) print("unsupport batch size {}".format(args.batch_size))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册