dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix Noise Reset Bug for SAC

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-01-15 13:28:40 +01:00
parent 844cdb1319
commit 43cc749809
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
metastable_baselines/sac/policies.py
View File

@ -389,7 +389,8 @@ class SACPolicy(BasePolicy):
:param batch_size: :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: def make_actor(self, features_extractor: Optional[BaseFeaturesExtractor] = None) -> Actor:
actor_kwargs = self._update_features_extractor( actor_kwargs = self._update_features_extractor(