final final fixes

This commit is contained in:
gerazov 2021-02-06 23:08:47 +01:00
parent ad17dc9e76
commit e507373b55
1 changed files with 2 additions and 1 deletions

View File

@ -35,7 +35,7 @@ def parse_arguments(argv):
"--continue_path", "--continue_path",
type=str, type=str,
help=("Training output folder to continue training. Used to continue " help=("Training output folder to continue training. Used to continue "
"a training. If it is used, "config_path" is ignored."), "a training. If it is used, 'config_path' is ignored."),
default="", default="",
required="--config_path" not in argv) required="--config_path" not in argv)
parser.add_argument( parser.add_argument(
@ -151,6 +151,7 @@ def process_args(args, model_type):
os.path.join(args.continue_path, "*.pth.tar") os.path.join(args.continue_path, "*.pth.tar")
) # * means all if need specific format then *.csv ) # * means all if need specific format then *.csv
args.restore_path = max(list_of_files, key=os.path.getctime) args.restore_path = max(list_of_files, key=os.path.getctime)
# checkpoint number based continuing
# args.restore_path = get_last_checkpoint(args.continue_path) # args.restore_path = get_last_checkpoint(args.continue_path)
print(f" > Training continues for {args.restore_path}") print(f" > Training continues for {args.restore_path}")