Dose using CPU faster than GPU?

vita-epfl / CrowdNav

[ICRA19] Crowd-aware Robot Navigation with Attention-based Deep Reinforcement Learning

MIT License

585 stars 169 forks source link

Dose using CPU faster than GPU? #25

Closed xinkai-jiang closed 4 years ago

xinkai-jiang commented 4 years ago

Hi, I try to train in RTX 2080 but it is too slow and 5000 episodes cost about 18h. The article told that train in CPU just took 10h. I wonder whether the CPU training is the recommanded option. Thank you!

ChanganVR commented 4 years ago

Hi John,

Yes, as I mentioned in some previous posts. This code runs faster on CPU probably because of the CPU-intensitve simulation.