don't save model without training

40acf6b5 · Chi-Liang Liu · Julien Chaumond · 189ff9b6 · 40acf6b5
Commit 40acf6b5 authored Jul 30, 2019 by Chi-Liang Liu Committed by Julien Chaumond Aug 18, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/run_squad.py examples/run_squad.py +1 -1

No files found.
--- a/examples/run_squad.py
+++ b/examples/run_squad.py
@@ -481,7 +481,7 @@ def main():


    # Save the trained model and the tokenizer
-    if args.local_rank == -1 or torch.distributed.get_rank() == 0:
+    if args.do_train and args.local_rank == -1 or torch.distributed.get_rank() == 0:
        # Create output directory if needed
        if not os.path.exists(args.output_dir) and args.local_rank in [-1, 0]:
            os.makedirs(args.output_dir)