dodox/PriorConditionedAnnealing
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed Bug: Wrong Params for instantiation of StdNet

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-05-21 17:08:45 +02:00
parent 224fc82a48
commit 852e1817f7
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
priorConditionedAnnealing/pca.py
View File

@ -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