Fix config init in `process_args`

This commit is contained in:
Eren Gölge 2021-06-29 16:41:08 +02:00
parent 4b5421b42f
commit 5723eb4738
2 changed files with 12 additions and 12 deletions

View File

@ -1012,17 +1012,18 @@ def process_args(args, config=None):
args.restore_path, best_model = get_last_checkpoint(args.continue_path) args.restore_path, best_model = get_last_checkpoint(args.continue_path)
if not args.best_path: if not args.best_path:
args.best_path = best_model args.best_path = best_model
# init config # init config if not already defined
if config is None and args.config_path: if config is None:
# init from a file if args.config_path:
config = load_config(args.config_path) # init from a file
else: config = load_config(args.config_path)
# init from console args else:
from TTS.config.shared_configs import BaseTrainingConfig # pylint: disable=import-outside-toplevel # init from console args
from TTS.config.shared_configs import BaseTrainingConfig # pylint: disable=import-outside-toplevel
config_base = BaseTrainingConfig() config_base = BaseTrainingConfig()
config_base.parse_known_args(coqpit_overrides) config_base.parse_known_args(coqpit_overrides)
config = register_config(config_base.model)() config = register_config(config_base.model)()
# override values from command-line args # override values from command-line args
config.parse_known_args(coqpit_overrides, relaxed_parser=True) config.parse_known_args(coqpit_overrides, relaxed_parser=True)
if config.mixed_precision: if config.mixed_precision:

View File

@ -1,8 +1,7 @@
import os import os
from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.vocoder.configs import HifiganConfig from TTS.vocoder.configs import HifiganConfig
from TTS.trainer import init_training, Trainer, TrainingArgs
output_path = os.path.dirname(os.path.abspath(__file__)) output_path = os.path.dirname(os.path.abspath(__file__))
config = HifiganConfig( config = HifiganConfig(