From 703be0499342c650aa09543107b26f4683af7eff Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Wed, 12 Dec 2018 12:02:10 +0100
Subject: [PATCH] bug fix

---
 layers/tacotron.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/layers/tacotron.py b/layers/tacotron.py
index 3e82486f..b972ed4a 100644
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@@ -338,8 +338,8 @@ class Decoder(nn.Module):
             gain=torch.nn.init.calculate_gain('linear'))
 
     def _reshape_memory(self, memory):
-        B = memory.shape[0]
         if memory is not None:
+            B = memory.shape[0]
             # Grouping multiple frames if necessary
             if memory.size(-1) == self.memory_dim:
                 memory = memory.contiguous()