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

Update SAC defaults

Browse source
This commit is contained in:
Antonin Raffin 2019-09-25 17:07:54 +02:00
parent 0e4fc9c0ac
commit 70e5de1d1b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/sac/sac.py
View file

@ -53,7 +53,7 @@ class SAC(BaseRLModel):
def __init__(self, policy, env, learning_rate=3e-4, buffer_size=int(1e6),
learning_starts=100, batch_size=64,
tau=0.005, ent_coef='auto', target_update_interval=1,
train_freq=-1, gradient_steps=-1, n_episodes_rollout=1,
train_freq=1, gradient_steps=1, n_episodes_rollout=-1,
target_entropy='auto', action_noise=None,
gamma=0.99, action_noise_std=0.0, create_eval_env=False,
policy_kwargs=None, verbose=0, seed=0, device='auto',