A code issue

hi, thanks for your open-source code. I met a bug when I run the beam search code in utilis/beamsearch.py line96: self.mask = self.mask.gather(1, perm_mask) where ‘perm_mask’ should be a LongTensor type, but here its type is FloatTensor. It makes severl beam search-based functions fail to use.

[chaitjo]

Here are some related issues and discussions for reference:

• https://github.com/chaitjo/graph-convnet-tsp/issues/11
• https://github.com/chaitjo/learning-tsp/issues/1#issuecomment-814688655

Maybe one simple thing to try first to get the code to run would be to update backpointers via integer division:

# Update backpointers
prev_k = bestScoresId // self.num_nodes

[chaitjo]

Great, happy to help, no worries.