Internal change

PiperOrigin-RevId: 306136233

Internal change
PiperOrigin-RevId: 306136233
7b8024c2 · A. Unique TensorFlower · 7194a5cc · 7b8024c2
Commit 7b8024c2 authored Apr 12, 2020 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

official/nlp/xlnet/xlnet_modeling.py official/nlp/xlnet/xlnet_modeling.py +2 -1

No files found.
--- a/official/nlp/xlnet/xlnet_modeling.py
+++ b/official/nlp/xlnet/xlnet_modeling.py
@@ -1019,7 +1019,8 @@ class Summarization(tf.keras.layers.Layer):
      summary = inputs[0]
    else:
      raise ValueError('Invalid summary type provided: %s' % self.summary_type)
-    summary = self.proj_layer(summary)
+    if self.use_proj:
+      summary = self.proj_layer(summary)
    summary = self.dropout_layer(summary)
    return summary