From 4fd2055f817e5b658c73e3b1b42cdfa48e929e13 Mon Sep 17 00:00:00 2001 From: vBarbaros Date: Mon, 15 Aug 2022 08:46:33 -0400 Subject: [PATCH] Remove obsolete params for SAC Agent constructor --- ReinforcementLearning/PolicyGradient/SAC/sac_torch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ReinforcementLearning/PolicyGradient/SAC/sac_torch.py b/ReinforcementLearning/PolicyGradient/SAC/sac_torch.py index 5117b57..01e2412 100644 --- a/ReinforcementLearning/PolicyGradient/SAC/sac_torch.py +++ b/ReinforcementLearning/PolicyGradient/SAC/sac_torch.py @@ -8,7 +8,7 @@ class Agent(): def __init__(self, alpha=0.0003, beta=0.0003, input_dims=[8], env=None, gamma=0.99, n_actions=2, max_size=1000000, tau=0.005, - layer1_size=256, layer2_size=256, batch_size=256, reward_scale=2): + batch_size=256, reward_scale=2): self.gamma = gamma self.tau = tau self.memory = ReplayBuffer(max_size, input_dims, n_actions)