Adding should_continue check for retraining (#3509)
This commit is contained in:
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user