dodox/metastable-baselines2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixing more issues with PCA

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-21 23:54:11 +02:00
parent c9ca5c0b53
commit 81e223183a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sbBrix/common/policies.py
View File

@ -676,6 +676,8 @@ class ActorCriticPolicy(BasePolicy):
return self.action_dist.proba_distribution(action_logits=mean_actions)
elif isinstance(self.action_dist, StateDependentNoiseDistribution):
return self.action_dist.proba_distribution(mean_actions, self.log_std, latent_pi)
elif isinstance(self.action_dist, PCA_Distribution):
return self.action_dist.proba_distribution(mean_actions, self.log_std)
else:
raise ValueError("Invalid action distribution")