Adding should_continue check for retraining (#3509)

2020-04-02 11:07:08 -07:00
parent 390c128592
commit 1b10159950
1 changed files with 1 additions and 0 deletions
--- a/examples/run_language_modeling.py
+++ b/examples/run_language_modeling.py
@@ -624,6 +624,7 @@ def main():
        and os.listdir(args.output_dir)
        and args.do_train
        and not args.overwrite_output_dir
+        and not args.should_continue
    ):
        raise ValueError(
            "Output directory ({}) already exists and is not empty. Use --overwrite_output_dir to overcome.".format(