From 8ef4b9c14364f64ef5b16a7a82e4735d1579e9cb Mon Sep 17 00:00:00 2001 From: MRXLT Date: Tue, 31 Mar 2020 06:51:17 +0000 Subject: [PATCH] fix criteo_ctr_with_cube/benchmark_batch.py --- python/examples/criteo_ctr_with_cube/benchmark_batch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/examples/criteo_ctr_with_cube/benchmark_batch.py b/python/examples/criteo_ctr_with_cube/benchmark_batch.py index f395d327..df5c6b90 100755 --- a/python/examples/criteo_ctr_with_cube/benchmark_batch.py +++ b/python/examples/criteo_ctr_with_cube/benchmark_batch.py @@ -56,7 +56,7 @@ def single_func(idx, resource): feed_dict["embedding_{}.tmp_0".format(i - 1)] = data[0][ i] feed_batch.append(feed_dict) - result = client.batch_predict(feed=feed_batch, fetch=fetch) + result = client.predict(feed=feed_batch, fetch=fetch) else: print("unsupport batch size {}".format(args.batch_size)) -- GitLab