Closed josch closed 4 years ago
Downgrading torch to version 0.4.1.post2 fixed this problem.
Please adapt superpoint_graph so that it works with the latest version of torch.
Thanks!
Hi,
for some reason torch.nn._functions.thnn.rnnFusedPointwise.GRUFused
does not exist anymore in 1.0.
A quick fix would simply be to change line 108 of learning/modules.py
from:
if input.is_cuda:
to
if input.is_cuda and torch.__version__.split('.')[0]=='0' :
Looking into a better fix for next update.
This is probably related to https://github.com/pytorch/pytorch/pull/10305 , so my suggestion would be:
cuda_fused_gru_cell = torch._C._VariableFunctions._thnn_fused_gru_cell
return cuda_fused_lstm_cell(gi, gh, hidden, self.bias_ih, self.bias_hh)
I haven't tried it myself, though.
New commit 7ac66d9e960f84014100bf087b996aa5dc9342c7 fixes the cpu version of GRUFused
Hi,
after following the instructions in
README.md
and runningI get the following output: