stable-baselines3-contrib-sacd/sb3_contrib/common/recurrent
Quentin Gallouédec 9cf8b5076f
Construct tensors directly on GPUs (#128)
* `to(device)` to `device=device` and `float()` to `dtype=th.float32`

* Update changelog

* Fix type checking

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2022-12-23 00:44:25 +01:00
..
__init__.py Recurrent PPO (#53) 2022-05-30 04:31:12 +02:00
buffers.py Fix reshape LSTM states (#112) 2022-10-26 18:03:45 +02:00
policies.py Construct tensors directly on GPUs (#128) 2022-12-23 00:44:25 +01:00
type_aliases.py Fix `sb3_contrib/common/recurrent/type_aliases.py` type hint (#121) 2022-11-29 10:41:07 +01:00