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

Fix typo: FrobeniusProjectionLay

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-01-16 15:27:20 +01:00
parent 0806a24036
commit a8930a897a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
metastable_baselines2/common/policies.py
View File

@ -37,7 +37,7 @@ from stable_baselines3.common.policies import ContinuousCritic
from stable_baselines3.common.type_aliases import Schedule, RolloutBufferSamples
from stable_baselines3.common.utils import get_device, is_vectorized_observation, obs_as_tensor
from metastable_projections.projections import BaseProjectionLayer, IdentityProjectionLayer, FrobeniusProjectionLay, WassersteinProjectionLayer, KLProjectionLayer
from metastable_projections.projections import BaseProjectionLayer, IdentityProjectionLayer, FrobeniusProjectionLayer, WassersteinProjectionLayer, KLProjectionLayer
from .distributions import make_proba_distribution