From d97b679513ea17f0d88a4da8eb4c0fcd62589d8a Mon Sep 17 00:00:00 2001 From: wangguibao Date: Fri, 22 Nov 2019 20:32:33 +0800 Subject: [PATCH] Fix embedding table dims --- elastic-ctr/client/demo/demo.cpp | 2 +- elastic-ctr/client/demo/elastic_ctr.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/elastic-ctr/client/demo/demo.cpp b/elastic-ctr/client/demo/demo.cpp index 2553697f..59a094ee 100644 --- a/elastic-ctr/client/demo/demo.cpp +++ b/elastic-ctr/client/demo/demo.cpp @@ -30,7 +30,7 @@ DEFINE_int32(batch_size, 10, "Infernce batch_size"); DEFINE_string(test_file, "", "test file"); const int VARIABLE_NAME_LEN = 256; -const int CTR_EMBEDDING_TABLE_SIZE = 400000001; +const int CTR_EMBEDDING_TABLE_SIZE = 100000001; struct Sample { std::map> slots; diff --git a/elastic-ctr/client/demo/elastic_ctr.py b/elastic-ctr/client/demo/elastic_ctr.py index abd53ff2..30873437 100644 --- a/elastic-ctr/client/demo/elastic_ctr.py +++ b/elastic-ctr/client/demo/elastic_ctr.py @@ -22,7 +22,7 @@ from elastic_ctr_api import ElasticCTRAPI BATCH_SIZE = 3 SERVING_IP = "127.0.0.1" SLOT_CONF_FILE = "./conf/slot.conf" -CTR_EMBEDDING_TABLE_SIZE = 400000001 +CTR_EMBEDDING_TABLE_SIZE = 100000001 SLOTS = [] -- GitLab