Use cpu instead of gpu - Githubissues

nachiket92 / conv-social-pooling

Code for model proposed in: Nachiket Deo and Mohan M. Trivedi,"Convolutional Social Pooling for Vehicle Trajectory Prediction." CVPRW, 2018

MIT License

326 stars 97 forks source link

Closed amirrezahjr closed 4 years ago

amirrezahjr commented 4 years ago

I don't have NVIDIA graphic installed on my system. I'm wondering if I can run this code with a limited dataset on CPU instead of GPU?

nachiket92 commented 4 years ago

You can run training on CPU if you set args['use_cuda'] to false in train.py. It will be much slower though.