From a7cbc660e051a757b311f7647624b030efa96d54 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 22 Aug 2023 01:12:50 +0200 Subject: [PATCH] Fix: Passed wrong args to pca propa net --- sbBrix/common/policies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sbBrix/common/policies.py b/sbBrix/common/policies.py index 9e06244..a01c431 100644 --- 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