采用稀疏特征emb后训练没有存储稀疏emb层参数,预测报错
Created by: menghou
def word_dis_emb(word, dict_size=DICT_SIZE, embed_size=EMBED_SIZE): """ """ embed = fluid.layers.embedding( input=word, size=[dict_size, embed_size], dtype='float32', param_attr=fluid.ParamAttr( name="dis_emb", learning_rate=5, initializer=fluid.initializer.Xavier(fan_in=embed_size, fan_out=embed_size) #initializer=fluid.initializer.Normal(scale=1/math.sqrt(dict_size)) ), is_sparse=True, is_distributed=True) return embed