search

devendrachaplot / DeepRL-Grounding

Train an RL agent to execute natural language instructions in a 3D Environment (PyTorch)
https://sites.google.com/view/gated-attention/home
MIT License
237 stars 37 forks source link

Can GPU speed up? #8

Closed TianyuanYu closed 6 years ago

TianyuanYu commented 6 years ago

If put the model in gpu, would it be faster? Would the model still be stable after hours?

Thanks.

devendrachaplot commented 6 years ago

We haven't tested training on a GPU, the model should behave similarly on a GPU.