Merge pull request #102 from Yoctol/pass-dropout-in-cls

Fix emb dropout by sending dropout in cls
Yoctol · Jun 10, 2019 · 36e7ab1 · 36e7ab1
2 parents f79facc + e155414
commit 36e7ab1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/talos/layers/embeddings.py b/talos/layers/embeddings.py
@@ -101,6 +101,7 @@ def from_weights(
             mask_index: Union[int, Sequence[int]] = None,
             constant: bool = False,
             auxiliary_tokens: int = 0,
+            dropout: float = None,
             **kwargs,
         ):
         '''Create a Embedding Layer by pre-defined matrix of shape (vocab_size, dimension).
@@ -134,6 +135,7 @@ def from_weights(
             embeddings_initializer=initializer,
             mask_index=mask_index,
             dtype=dtype,
+            dropout=dropout,
             **kwargs,
         )
         if constant: