Revert softmax deprecation fix till new release

experimenti · Oct 30, 2017 · 098e0dc · 098e0dc
1 parent 77ef1d8
commit 098e0dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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_