未验证 提交 45f9c9eb 编写于 作者: Z zhangkaihuo 提交者: GitHub

recovery code (#39287)

上级 ffa7ff9c
...@@ -75,7 +75,7 @@ class FusedMultiHeadAttention(Layer): ...@@ -75,7 +75,7 @@ class FusedMultiHeadAttention(Layer):
embed_dim, embed_dim,
num_heads, num_heads,
dropout_rate=0.5, dropout_rate=0.5,
attn_dropout_rate=None, attn_dropout_rate=0.5,
kdim=None, kdim=None,
vdim=None, vdim=None,
normalize_before=False, normalize_before=False,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册