remove unnecessary f0 loss in discrimnator

2026-02-05 01:49:25 +08:00 · 2024-10-18 15:58:19 +08:00
parent a4db3db8ed
commit 5bd5dfecab
3 changed files with 27 additions and 27 deletions
--- a/cosyvoice/bin/train.py
+++ b/cosyvoice/bin/train.py
@@ -95,6 +95,8 @@ def main():
        override_dict.pop('hift')
    with open(args.config, 'r') as f:
        configs = load_hyperpyyaml(f, overrides=override_dict)
+    if gan is True:
+        configs['train_conf'] = configs['train_conf_gan']
    configs['train_conf'].update(vars(args))

    # Init env for ddp