mirror of https://github.com/coqui-ai/TTS.git
Fix config init in `process_args`
This commit is contained in:
parent
4b5421b42f
commit
5723eb4738
|
@ -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:
|
||||||
|
|
|
@ -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(
|
||||||
|
|
Loading…
Reference in New Issue