From 43cc7498090e736c6b0116d97e88f1c600fafb2e Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 15 Jan 2023 13:28:40 +0100 Subject: [PATCH] Fix Noise Reset Bug for SAC --- metastable_baselines/sac/policies.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/metastable_baselines/sac/policies.py b/metastable_baselines/sac/policies.py index 3a6b297..9540123 100644 --- a/metastable_baselines/sac/policies.py +++ b/metastable_baselines/sac/policies.py @@ -389,7 +389,8 @@ class SACPolicy(BasePolicy): :param batch_size: """ - self.actor.reset_noise(batch_size=batch_size) + # self.actor.reset_noise(batch_size=batch_size) + self.actor.reset_noise() def make_actor(self, features_extractor: Optional[BaseFeaturesExtractor] = None) -> Actor: actor_kwargs = self._update_features_extractor(