From 852e1817f7a90904240e92039d0b85d95f29f957 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 21 May 2023 17:08:45 +0200 Subject: [PATCH] Fixed Bug: Wrong Params for instantiation of StdNet --- priorConditionedAnnealing/pca.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/priorConditionedAnnealing/pca.py b/priorConditionedAnnealing/pca.py index 1454e53..e37e231 100644 --- a/priorConditionedAnnealing/pca.py +++ b/priorConditionedAnnealing/pca.py @@ -118,8 +118,7 @@ class PCA_Distribution(SB3_Distribution): def proba_distribution_net(self, latent_dim: int): mu_net = nn.Linear(latent_dim, self.action_dim) - std_net = StdNet(latent_dim, self.action_dim, - self.init_std, self.par_strength) + std_net = StdNet(latent_dim, self.action_dim, self.init_std, self.par_strength, self.epsilon) return mu_net, std_net