From 6478d552dcf0ee4a75f7acd9745fc138ff1bb2ef Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Sat, 12 Dec 2020 23:58:38 +0100
Subject: [PATCH] tacotron training bug fix

---
 TTS/bin/train_tacotron.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/TTS/bin/train_tacotron.py b/TTS/bin/train_tacotron.py
index da54831a..f75b44af 100644
--- a/TTS/bin/train_tacotron.py
+++ b/TTS/bin/train_tacotron.py
@@ -604,7 +604,9 @@ def main(args):  # pylint: disable=redefined-outer-name
             if c.bidirectional_decoder:
                 model.decoder_backward.set_r(r)
             train_loader.dataset.outputs_per_step = r
+            train_loader.batch_size = c.batch_size
             eval_loader.dataset.outputs_per_step = r
+            eval_loader.batch_size = c.batch_size
             print("\n > Number of output frames:", model.decoder.r)
         train_avg_loss_dict, global_step = train(train_loader, model,
                                                  criterion, optimizer,