stable-baselines3-contrib-sacd/sb3_contrib
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
..
ars Construct tensors directly on GPUs (#128) 2022-12-23 00:44:25 +01:00
common Construct tensors directly on GPUs (#128) 2022-12-23 00:44:25 +01:00
ppo_mask Expose modules in `__init__.py` with `__all__` attribute (#124) 2022-12-05 15:53:57 +01:00
ppo_recurrent Construct tensors directly on GPUs (#128) 2022-12-23 00:44:25 +01:00
qrdqn [Feature] Non-shared features extractor in on-policy algorithms (#130) 2022-12-23 00:23:45 +01:00
tqc Construct tensors directly on GPUs (#128) 2022-12-23 00:44:25 +01:00
trpo Expose modules in `__init__.py` with `__all__` attribute (#124) 2022-12-05 15:53:57 +01:00
__init__.py Expose modules in `__init__.py` with `__all__` attribute (#124) 2022-12-05 15:53:57 +01:00
py.typed Add TQC and base scripts 2020-09-25 12:47:45 +02:00
version.txt [Feature] Non-shared features extractor in on-policy algorithms (#130) 2022-12-23 00:23:45 +01:00