mirror of https://github.com/coqui-ai/TTS.git
final final fixes
This commit is contained in:
parent
ad17dc9e76
commit
e507373b55
|
@ -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}")
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue