疏忽

ypw757 · Aug 11, 2022 · 0cd0a4a · 0cd0a4a
1 parent 35d762a
commit 0cd0a4a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bert4keras/layers.py b/bert4keras/layers.py
@@ -606,6 +606,7 @@ def __init__(
         self.activation = activations.get(activation)
         self.use_bias = use_bias
         self.normalization = normalization
+        self.self_attention = self_attention
         self.attention_scale = attention_scale
         self.attention_dropout = attention_dropout
         self.kernel_initializer = initializers.get(kernel_initializer)