Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit 422aac7

Browse files
committed
trainer-utils: do not show input sentence in decoded file
1 parent 668e385 commit 422aac7

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tensor2tensor/utils/trainer_utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -703,8 +703,7 @@ def log_fn(inputs, outputs):
703703
tf.logging.info("Writing decodes into %s" % decode_filename)
704704
outfile = tf.gfile.Open(decode_filename, "w")
705705
for index in range(len(sorted_inputs)):
706-
outfile.write("%s\t%s\n" % (decodes[sorted_keys[index]],
707-
sorted_inputs[sorted_keys[index]]))
706+
outfile.write("%s\n" % (decodes[sorted_keys[index]]))
708707

709708

710709
def decode_interactively(estimator):

0 commit comments

Comments
 (0)