imitation-learning icon indicating copy to clipboard operation
imitation-learning copied to clipboard

About GPU

Open gnniaaa opened this issue 3 years ago • 0 comments

When I try to run train.py, I find it run with CPU instead of GPU, how can I use GPU?

gnniaaa avatar Apr 16 '21 15:04 gnniaaa