chaitjo/graph-convnet-tsp

RuntimeError when executing "test backward pass" block in main.ipynb

bluesquinn777 opened this issue · 7 comments

Hi, when I directly run your code in main.ipynb, other parts are correct, but I received "RuntimeError: grad_input must be contiguous"

image

I think this may be caused by loss_edges function, where y = y.permute(0, 3, 1, 2) which makes y not contiguous anymore? Thanks!

I had the same problem at runtime.

Did you fix this discontiguous problem?

Thanks for your interest - please make sure that you use the correct versions of Cuda and PyTorch as prescribed in the README.

Thanks for your interest - please make sure that you use the correct versions of Cuda and PyTorch as prescribed in the README.

We ran our code on Ubuntu 16.04, using Python 3.6.7, PyTorch 0.4.1 and CUDA 9.0.

Thanks for your reply, I'm using windows, Python 3. CUDA 11.3, PyTorch 1.11.0, which may lead to this problem. Add .contiguous() after permute in the loss function can solve this problem.

I configured it according to the README, and it is running successfully now, thanks to the author.

Great

.contiguous()

where do i need to add this tks ! @bluesquinn777