saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-04 04:07:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Bump version and change default noise clipping

Browse source
This commit is contained in:
Antonin Raffin 2019-11-08 14:59:42 +01:00
parent 715865a0fe
commit f2a61949ae
3 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
setup.py
View file

@ -34,7 +34,7 @@ setup(name='torchy_baselines',
license="MIT",
long_description="",
long_description_content_type='text/markdown',
version="0.0.4",
version="0.0.6a",
)
# python setup.py sdist

2
torchy_baselines/__init__.py
View file

@ -4,4 +4,4 @@ from torchy_baselines.ppo import PPO
from torchy_baselines.sac import SAC
from torchy_baselines.td3 import TD3
__version__ = "0.0.5a"
__version__ = "0.0.6a"

4
torchy_baselines/td3/policies.py
View file

@ -7,7 +7,7 @@ from torchy_baselines.common.policies import BasePolicy, register_policy, create
class Actor(BaseNetwork):
def __init__(self, obs_dim, action_dim, net_arch, activation_fn=nn.ReLU,
use_sde=False, log_std_init=-2, clip_noise=0.1):
use_sde=False, log_std_init=-2, clip_noise=0.5):
super(Actor, self).__init__()
self.latent_pi, self.log_std = None, None
@ -67,7 +67,7 @@ class Critic(BaseNetwork):
class TD3Policy(BasePolicy):
def __init__(self, observation_space, action_space,
learning_rate, net_arch=None, device='cpu',
activation_fn=nn.ReLU, use_sde=False, log_std_init=-2, clip_noise=0.1):
activation_fn=nn.ReLU, use_sde=False, log_std_init=-2, clip_noise=0.5):
super(TD3Policy, self).__init__(observation_space, action_space, device)
if net_arch is None: