fixed to newer torch7

ketranm · Jan 31, 2019 · 7ba85fb · 7ba85fb
1 parent 9b1e56c
commit 7ba85fb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/Emission.lua b/Emission.lua
@@ -35,6 +35,9 @@ function model:log_prob(input)
         self._logp = self._cache
     end
 
+    if input:type() == 'torch.IntTensor' then
+        input = input:long()
+    end
     return self._logp:index(2, input:view(-1)):view(-1, N, T):transpose(1, 2):transpose(2, 3)
 end
 

diff --git a/main.lua b/main.lua
@@ -273,8 +273,9 @@ function infer(textfile, predfile, modelfile)
             local x = sents[k+i]
             input[{{k+1}, {1, x:numel()}}] = x
         end
-
-        input = input:cuda()
+        if opt.cuda then
+          input = input:cuda()
+        end
         io.write(string.format('sent: %d\r', i))
         io.flush()
         local log_prior = prior_net:log_prob(input)