Internal change

PiperOrigin-RevId: 311428193

Internal change
PiperOrigin-RevId: 311428193
3c5d118c · Chen Chen · A. Unique TensorFlower · 460943f9 · 3c5d118c
Commit 3c5d118c authored May 13, 2020 by Chen Chen Committed by A. Unique TensorFlower May 13, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

official/nlp/modeling/networks/encoder_scaffold.py official/nlp/modeling/networks/encoder_scaffold.py +2 -1

No files found.
--- a/official/nlp/modeling/networks/encoder_scaffold.py
+++ b/official/nlp/modeling/networks/encoder_scaffold.py
@@ -116,7 +116,8 @@ class EncoderScaffold(tf.keras.Model):

    if embedding_cls:
      if inspect.isclass(embedding_cls):
-        self._embedding_network = embedding_cls(embedding_cfg)
+        self._embedding_network = embedding_cls(
+            **embedding_cfg) if embedding_cfg else embedding_cls()
      else:
        self._embedding_network = embedding_cls
      inputs = self._embedding_network.inputs