Fix: Passed wrong args to pca propa net

2023-08-22 01:12:50 +02:00 · 2023-08-22 01:12:50 +02:00 · a7cbc660e0
commit a7cbc660e0
parent 0af83eb7dc
1 changed files with 1 additions and 1 deletions
--- a/sbBrix/common/policies.py
+++ b/sbBrix/common/policies.py
@ -815,7 +815,7 @@ class Actor(BasePolicy):
                action_dim, **dist_kwargs
            )
            self.mu, self.log_std = self.action_dist.proba_distribution_net(
-                latent_dim=last_layer_dim, return_log_std=True, **dist_kwargs
+                latent_dim=last_layer_dim, return_log_std=True
            )
            self._remember_log_std = log_std_init
            # Avoid numerical issues by limiting the mean of the Gaussian