wouterkool / attention-learn-to-route

Attention based model for learning to solve different routing problems
MIT License
1.04k stars 337 forks source link

Running time #45

Closed SwiderCharlie closed 2 years ago

SwiderCharlie commented 2 years ago

Hello! I ran the code with the following command: python run.py --graph_size 20 --baseline rollout --run_name 'tsp20_rollout' --val_dataset data/tsp/tsp20_validation_seed4321.pkl but an epoch took nearly 3 hours. I ran the code on a single GPU(Nvidia Tesla V100S ) and I verified that opts.device is indeed cuda:0. What is the reason for this? Thanks in advance!