From 098e0dcf9b18394b34fdb33e2117ca382fcff0fe Mon Sep 17 00:00:00 2001
From: SeanNaren <seannaren@hotmail.com>
Date: Mon, 30 Oct 2017 13:11:34 +0000
Subject: [PATCH] Revert softmax deprecation fix till new release

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index b228fd81..e7b5ad64 100644
--- a/model.py
+++ b/model.py
@@ -41,7 +41,7 @@ class InferenceBatchLogSoftmax(nn.Module):
     def forward(self, input_):
         if not self.training:
             batch_size = input_.size()[0]
-            return torch.stack([F.log_softmax(input_[i], dim=1) for i in range(batch_size)], 0)
+            return torch.stack([F.log_softmax(input_[i]) for i in range(batch_size)], 0)
         else:
             return input_