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(