Update lstm.py

Fix original issue huggingface#34
MikhailKazekin · Jun 19, 2020 · 9771c88 · 9771c88
1 parent 198f7d4
commit 9771c88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmoji/lstm.py b/torchmoji/lstm.py
@@ -75,7 +75,7 @@ def reset_parameters(self):
     def forward(self, input, hx=None):
         is_packed = isinstance(input, PackedSequence)
         if is_packed:
-            input, batch_sizes = input
+            input, batch_sizes, _, _ = input
             max_batch_size = batch_sizes[0]
         else:
             batch_sizes = None