From ae9a95cbfc25aa2c364a557056d348cba03602c2 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Thu, 2 Feb 2023 20:00:16 +0100 Subject: [PATCH] Fix: Typo in SAC --- metastable_baselines/sac/policies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metastable_baselines/sac/policies.py b/metastable_baselines/sac/policies.py index a01d64c..010d5c8 100644 --- a/metastable_baselines/sac/policies.py +++ b/metastable_baselines/sac/policies.py @@ -320,7 +320,7 @@ class SACPolicy(BasePolicy): } self.actor_kwargs.update(sde_kwargs) - self.actor.kwargs.update({'latent_dim_sde': latent_dim_sde}) + self.actor_kwargs.update({'latent_dim_sde': latent_dim_sde}) self.critic_kwargs = self.net_args.copy() self.critic_kwargs.update( {