stable-baselines3/stable_baselines3
Quentin Gallouédec 68a40e0940
Construct tensors directly on GPU (#1218)
* Replace .to(device) when possible

* fix numpy dep

* black

* Add warning for device != cpu and copy=False

* Update changelog

* Remove warning

* Update buffers.py
2022-12-19 12:50:22 +01:00
..
a2c Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
common Construct tensors directly on GPU (#1218) 2022-12-19 12:50:22 +01:00
ddpg Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
dqn Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
her Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
ppo Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
sac Construct tensors directly on GPU (#1218) 2022-12-19 12:50:22 +01:00
td3 Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
__init__.py Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00
py.typed Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
version.txt Exposed modules in __init__.py with __all__ (#1195) 2022-11-29 23:33:46 +01:00