Open th-yoon opened 3 years ago
I fixed this issue by commenting out the __getitem__
overrides of the following classes:
attention_model/AttentionModelFixed
and state_tsp/StateTSP
Which version of Python are you using? I think this is related to using <3.8. Please make sure you use 3.8. Commenting out __getitem__
probably works fine as long as you're not using beam search (which requires indexing into the state array).
@wouterkool Thanks for your quick reply. Current python version is 3.7.9. Will try with python >= 3.8.
Command:
python3 eval.py data/tsp/tsp20_validation_seed4321.pkl --model pretrained/tsp_20 --decode_strategy greedy
Trace: