hybrid wide branch

8964faed · ShawnXuan · d06761bd · 8964faed
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

ClickThroughRate/WideDeepLearning/wdl_train_eval_with_hybrid_embd.py ...hRate/WideDeepLearning/wdl_train_eval_with_hybrid_embd.py +3 -3

未找到文件。
--- a/ClickThroughRate/WideDeepLearning/wdl_train_eval_with_hybrid_embd.py
+++ b/ClickThroughRate/WideDeepLearning/wdl_train_eval_with_hybrid_embd.py
@@ -177,9 +177,9 @@ def _embedding(name, ids, embedding_size, vocab_size, split_axis=0):


 def _model(dense_fields, wide_sparse_fields, deep_sparse_fields):
-    wide_embedding = _embedding('wide_embedding', wide_sparse_fields, 1, FLAGS.wide_vocab_size)
-    # wide_embedding = _hybrid_embedding('wide_embedding', wide_sparse_fields, 1, FLAGS.wide_vocab_size, 
-    #                                    FLAGS.hf_wide_vocab_size)
+    # wide_embedding = _embedding('wide_embedding', wide_sparse_fields, 1, FLAGS.wide_vocab_size)
+    wide_embedding = _hybrid_embedding('wide_embedding', wide_sparse_fields, 1, FLAGS.wide_vocab_size, 
+                                       FLAGS.hf_wide_vocab_size)
    wide_scores = flow.math.reduce_sum(wide_embedding, axis=[1], keepdims=True)
    wide_scores = flow.parallel_cast(wide_scores, distribute=flow.distribute.split(0),
                                     gradient_distribute=flow.distribute.broadcast())