From f7195443864255541bae3b92ff169e6dda7d8ef3 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Thu, 14 Nov 2019 14:35:47 +0100 Subject: [PATCH] Change default batch size for SAC --- torchy_baselines/sac/sac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchy_baselines/sac/sac.py b/torchy_baselines/sac/sac.py index f4c5c34..5def28f 100644 --- a/torchy_baselines/sac/sac.py +++ b/torchy_baselines/sac/sac.py @@ -54,7 +54,7 @@ class SAC(BaseRLModel): :param _init_setup_model: (bool) Whether or not to build the network at the creation of the instance """ def __init__(self, policy, env, learning_rate=3e-4, buffer_size=int(1e6), - learning_starts=100, batch_size=64, + learning_starts=100, batch_size=256, tau=0.005, ent_coef='auto', target_update_interval=1, train_freq=1, gradient_steps=1, n_episodes_rollout=-1, target_entropy='auto', action_noise=None,