Parcourir la source

Set S2T as default finetuning model

Ruslan Mavlyutov il y a 2 ans
Parent
commit
08a8eddcda
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      scripts/m4t/finetune/finetune.py

+ 1 - 1
scripts/m4t/finetune/finetune.py

@@ -112,7 +112,7 @@ def init_parser() -> argparse.ArgumentParser:
         "--mode",
         type=trainer.FinetuneMode,
         choices=list(trainer.FinetuneMode),
-        default=trainer.FinetuneMode.TEXT_TO_SPEECH,
+        default=trainer.FinetuneMode.SPEECH_TO_TEXT,
         help=(
             "* `SPEECH_TO_SPEECH` -- finetune S2T and T2U parts of the model; "
             "* `TEXT_TO_SPEECH` -- finetune only T2U; "