提交 e041e650 编写于 作者: W wangguibao

Fix embedding table dims

上级 8325c59e
......@@ -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<std::string, std::vector<uint64_t>> slots;
......
......@@ -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 = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册