diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 4959d7ba..9dfdc067 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -1671,8 +1671,8 @@ class Vits(BaseTTS): Returns: List: Schedulers, one for each optimizer. """ - scheduler_G = get_scheduler(self.config.lr_scheduler_gen, self.config.lr_scheduler_gen_params, optimizer[0]) - scheduler_D = get_scheduler(self.config.lr_scheduler_disc, self.config.lr_scheduler_disc_params, optimizer[1]) + scheduler_D = get_scheduler(self.config.lr_scheduler_disc, self.config.lr_scheduler_disc_params, optimizer[0]) + scheduler_G = get_scheduler(self.config.lr_scheduler_gen, self.config.lr_scheduler_gen_params, optimizer[1]) return [scheduler_D, scheduler_G] def get_criterion(self):