embeddings check condition in seq2seq_model

google · Jul 10, 2017 · ab98771 · ab98771
1 parent 6403218
commit ab98771
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/seq2seq/models/seq2seq_model.py b/seq2seq/models/seq2seq_model.py
@@ -136,7 +136,7 @@ def batch_size(self, features, labels):
   def source_embedding(self):
     """Returns the embedding used for the source sequence.
     """
-    if self.embedding_mat:
+    if self.embedding_mat is not None:
       self.params.update({"embedding.dim":self.embedding_mat.shape[1]})
       initializer = tf.constant(self.embedding_mat, dtype=tf.float32)
       shape_ = None