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,8 +1012,9 @@ def process_args(args, config=None):
args.restore_path, best_model = get_last_checkpoint(args.continue_path)
if not args.best_path:
args.best_path = best_model
# init config
if config is None and args.config_path:
# init config if not already defined
if config is None:
if args.config_path:
# init from a file
config = load_config(args.config_path)
else:

View File

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