Error in beamsearch.py about graph-convnet-tsp HOT 6 CLOSED

It worked now.
Possibly some cuda issue.

Thanks :)

from graph-convnet-tsp.

Hi @sahilm1992, thanks for your interest! That's unexpected, but maybe you can fix it by simply converting the float tensor to a long tensor?

E.g. self.mask = self.mask.gather(1, perm_mask.long())

Let me know if that works.

from graph-convnet-tsp.

let me try that. Could you confirm if the implementation works on gpu?

from graph-convnet-tsp.

Yes, all experiments in the associated paper were run using this codebase. You can read about our exact hardware setup there. However, the code has not been updated since ~ 1 year. You can try the updated repository associated with the more comprehensive paper, too.

from graph-convnet-tsp.

By the way, can you tell me your hardware setup? E.g. Operating system, Cuda version, GPU type, etc.

from graph-convnet-tsp.

Great, feel free to open further issues if you need help.

from graph-convnet-tsp.