stable-baselines3/docs/modules
Devin White 56c153f048
Add warning when using PPO on GPU and update doc (#2017)
* Update documentation

Added comment to PPO documentation that CPU should primarily be used unless using CNN as well as sample code. Added warning to user for both PPO and A2C that CPU should be used if the user is running GPU without using a CNN, reference Issue #1245.

* Add warning to base class and add test

---------

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2024-10-07 11:24:47 +02:00
..
a2c.rst Update outdated custom env doc (#1490) 2023-05-08 13:48:26 +02:00
base.rst Review of code (A2C, PPO and refactoring) (#35) 2020-06-09 13:54:18 +02:00
ddpg.rst Update outdated custom env doc (#1490) 2023-05-08 13:48:26 +02:00
dqn.rst Update outdated custom env doc (#1490) 2023-05-08 13:48:26 +02:00
her.rst Update outdated custom env doc (#1490) 2023-05-08 13:48:26 +02:00
ppo.rst Add warning when using PPO on GPU and update doc (#2017) 2024-10-07 11:24:47 +02:00
sac.rst Update outdated custom env doc (#1490) 2023-05-08 13:48:26 +02:00
td3.rst Update outdated custom env doc (#1490) 2023-05-08 13:48:26 +02:00