search

facebookresearch / torchbeast

A PyTorch Platform for Distributed RL
Apache License 2.0
734 stars 113 forks source link

torchbeast RL lib from FAIR is giving INF loss on TPU in Colab PRO #21

Open den-run-ai opened 3 years ago

den-run-ai commented 3 years ago

Torchbeast RL lib from FAIR is giving INF loss on TPU in Colab PRO. Detailed report of this issue here: https://github.com/pytorch/xla/issues/2740