提交 d97b6795 编写于 作者: W wangguibao

Fix embedding table dims

上级 c6519410
...@@ -30,7 +30,7 @@ DEFINE_int32(batch_size, 10, "Infernce batch_size"); ...@@ -30,7 +30,7 @@ DEFINE_int32(batch_size, 10, "Infernce batch_size");
DEFINE_string(test_file, "", "test file"); DEFINE_string(test_file, "", "test file");
const int VARIABLE_NAME_LEN = 256; const int VARIABLE_NAME_LEN = 256;
const int CTR_EMBEDDING_TABLE_SIZE = 400000001; const int CTR_EMBEDDING_TABLE_SIZE = 100000001;
struct Sample { struct Sample {
std::map<std::string, std::vector<uint64_t>> slots; std::map<std::string, std::vector<uint64_t>> slots;
......
...@@ -22,7 +22,7 @@ from elastic_ctr_api import ElasticCTRAPI ...@@ -22,7 +22,7 @@ from elastic_ctr_api import ElasticCTRAPI
BATCH_SIZE = 3 BATCH_SIZE = 3
SERVING_IP = "127.0.0.1" SERVING_IP = "127.0.0.1"
SLOT_CONF_FILE = "./conf/slot.conf" SLOT_CONF_FILE = "./conf/slot.conf"
CTR_EMBEDDING_TABLE_SIZE = 400000001 CTR_EMBEDDING_TABLE_SIZE = 100000001
SLOTS = [] SLOTS = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册