Fix: convert CudaLongTensor to CudaTensor

Laulian · Aug 17, 2016 · cd3a8fc · cd3a8fc
1 parent adc6885
commit cd3a8fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.lua b/main.lua
@@ -1044,7 +1044,7 @@ function stereo_predict(x_batch, id)
       end
 
       _, d = torch.min(vol, 2)
-      disp[direction == 1 and 1 or 2] = d:add(-1)
+      disp[direction == 1 and 1 or 2] = d:cuda():add(-1)
    end
    collectgarbage()