Adding should_continue check for retraining (#3509)

This commit is contained in:
Mark Kockerbeck
2020-04-02 11:07:08 -07:00
committed by GitHub
parent 390c128592
commit 1b10159950

View File

@@ -624,6 +624,7 @@ def main():
and os.listdir(args.output_dir) and os.listdir(args.output_dir)
and args.do_train and args.do_train
and not args.overwrite_output_dir and not args.overwrite_output_dir
and not args.should_continue
): ):
raise ValueError( raise ValueError(
"Output directory ({}) already exists and is not empty. Use --overwrite_output_dir to overcome.".format( "Output directory ({}) already exists and is not empty. Use --overwrite_output_dir to overcome.".format(