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!
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 thatopts.device
is indeedcuda:0
. What is the reason for this? Thanks in advance!