提交 5bfb26a8 编写于 作者: J JiabinYang

test=develop, fix embeding distribute and sparse can't be true and the same time

上级 2e38faa3
...@@ -336,7 +336,7 @@ def embedding(input, ...@@ -336,7 +336,7 @@ def embedding(input,
""" """
helper = LayerHelper('embedding', **locals()) helper = LayerHelper('embedding', **locals())
remote_prefetch = is_sparse remote_prefetch = is_sparse and (not is_distributed)
if remote_prefetch: if remote_prefetch:
assert is_sparse is True and is_distributed is False assert is_sparse is True and is_distributed is False
w = helper.create_parameter( w = helper.create_parameter(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册