From 3253ee11e7a941946c3a252e63d981d6735df401 Mon Sep 17 00:00:00 2001 From: RaphaelWag <39496028+RaphaelWag@users.noreply.github.com> Date: Fri, 31 Jul 2020 11:10:48 +0200 Subject: [PATCH] Update custom_policy.rst (#125) * Update custom_policy.rst Fixed Typo * Update changelog.rst Co-authored-by: Antonin RAFFIN --- docs/guide/custom_policy.rst | 2 +- docs/misc/changelog.rst | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docs/guide/custom_policy.rst b/docs/guide/custom_policy.rst index 2ce9202..755baee 100644 --- a/docs/guide/custom_policy.rst +++ b/docs/guide/custom_policy.rst @@ -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) diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst index 7bb894f..16df2ab 100644 --- a/docs/misc/changelog.rst +++ b/docs/misc/changelog.rst @@ -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