riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

[run_seq2seq.py] restore functionality: saving to test_generations.txt (#10428)

Browse Source 
This PR restores the original functionality that for some reason was modified.

Fixes: https://github.com/huggingface/transformers/issues/10381

@sgugger
This commit is contained in:
Stas Bekman 2021-02-27 08:21:50 -08:00 committed by GitHub
parent 311b7048c5
commit f52a15897b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/seq2seq/run_seq2seq.py
View File

@ -629,7 +629,7 @@ def main():
test_results.predictions, skip_special_tokens=True, clean_up_tokenization_spaces=True test_results.predictions, skip_special_tokens=True, clean_up_tokenization_spaces=True
) )
test_preds = [pred.strip() for pred in test_preds] test_preds = [pred.strip() for pred in test_preds]
output_test_preds_file = os.path.join(training_args.output_dir, "test_preds_seq2seq.txt") output_test_preds_file = os.path.join(training_args.output_dir, "test_generations.txt")
with open(output_test_preds_file, "w") as writer: with open(output_test_preds_file, "w") as writer:
writer.write("\n".join(test_preds)) writer.write("\n".join(test_preds))