write attention debug to log file (OpenNMT#1384)

AMusketeer · Apr 5, 2019 · 19b52ec · 19b52ec
1 parent ad50970
commit 19b52ec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/onmt/translate/translator.py b/onmt/translate/translator.py
@@ -372,7 +372,10 @@ def translate(
                             "{:*>10.7f} ", "{:>10.7f} ", max_index)
                         output += row_format.format(word, *row) + '\n'
                         row_format = "{:>10.10} " + "{:>10.7f} " * len(srcs)
-                    os.write(1, output.encode('utf-8'))
+                    if self.logger:
+                        self.logger.info(output)
+                    else:
+                        os.write(1, output.encode('utf-8'))
 
         end_time = time.time()