mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-07-29 20:14:17 +00:00
Update custom_policy.rst (#125)
* Update custom_policy.rst Fixed Typo * Update changelog.rst Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
This commit is contained in:
parent
77cb3dd0ab
commit
3253ee11e7
2 changed files with 3 additions and 2 deletions
|
|
@ -16,7 +16,7 @@ using ``policy_kwargs`` parameter:
|
|||
|
||||
from stable_baselines3 import PPO
|
||||
|
||||
# Custom MLP policy of two layers of size 32 each with tanh activation function
|
||||
# Custom MLP policy of two layers of size 32 each with Relu activation function
|
||||
policy_kwargs = dict(activation_fn=th.nn.ReLU, net_arch=[32, 32])
|
||||
# Create the agent
|
||||
model = PPO("MlpPolicy", "CartPole-v1", policy_kwargs=policy_kwargs, verbose=1)
|
||||
|
|
|
|||
|
|
@ -57,6 +57,7 @@ Documentation:
|
|||
- Added Unity reacher to the projects page (@koulakis)
|
||||
- Added PyBullet colab notebook
|
||||
- Fixed typo in PPO example code (@joeljosephjin)
|
||||
- Fixed typo in custom policy doc (@RaphaelWag)
|
||||
|
||||
|
||||
|
||||
|
|
@ -359,4 +360,4 @@ And all the contributors:
|
|||
@Miffyli @dwiel @miguelrass @qxcv @jaberkow @eavelardev @ruifeng96150 @pedrohbtp @srivatsankrishnan @evilsocket
|
||||
@MarvineGothic @jdossgollin @SyllogismRXS @rusu24edward @jbulow @Antymon @seheevic @justinkterry @edbeeching
|
||||
@flodorner @KuKuXia @NeoExtended @PartiallyTyped @mmcenta @richardwu @kinalmehta @rolandgvc @tkelestemur @mloo3
|
||||
@tirafesi @blurLake @koulakis @joeljosephjin @shwang @rk37 @andyshih12
|
||||
@tirafesi @blurLake @koulakis @joeljosephjin @shwang @rk37 @andyshih12 @RaphaelWag
|
||||
|
|
|
|||
Loading…
Reference in a new issue