diff --git a/model.py b/model.py
index 5e7e103..f576350 100644
--- a/model.py
+++ b/model.py
@@ -42,7 +42,7 @@ def forward(self, x, length=None, writer=None):
             fc = self.fc(decoder)
             return fc.contiguous() if self.training else fc.contiguous(), [weight.contiguous() for weight in w]
         else:
-            return self.generate(self.Decoder, x, length, writer).contiguous()
+            return self.generate(self.Decoder, x, length, writer).contiguous().tolist()
 
     def generate(self, decode_fn, prior: torch.Tensor, length=2048, tf_board_writer: SummaryWriter = None):
         decode_array = prior