Fixes issue abisee#4 for NaN loss

pedrobalage · May 31, 2017 · 434d696 · 434d696
1 parent b3cae0e
commit 434d696
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/model.py b/model.py
@@ -217,8 +217,6 @@ def _calc_final_dist(self, vocab_dists, attn_dists):
             final_dists = [vocab_dist + copy_dist for (vocab_dist, copy_dist) in zip(
                 vocab_dists_extended, attn_dists_projected)]
 
-            return final_dists
-
             # Issue #4: suggestion from user: 'rahul-iisc'
             # OOV part of vocab is max_art_oov long. Not all the sequences in a batch will have max_art_oov tokens.
             # That will cause some entries to be 0 in the distribution, which will result in NaN when calulating log_dists

diff --git a/run_summarization.py b/run_summarization.py
@@ -31,7 +31,7 @@
 import util
 
 os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+os.environ["CUDA_VISIBLE_DEVICES"] = "1"
 
 FLAGS = tf.app.flags.FLAGS