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

Factor Projections out into metastable-projections

Browse Source
This commit is contained in:
dodox
2022-09-03 11:37:41 +02:00
parent 0a037deccc
commit 4bb772a251
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
metastable_baselines/ppo/policies.py
+1 -1
View File
@@ -37,7 +37,7 @@ from stable_baselines3.common.torch_layers import (
from stable_baselines3.common.preprocessing import get_action_dim
from metastable_baselines.projections.w2_projection_layer import WassersteinProjectionLayer
from metastable_projections.projections.w2_projection_layer import WassersteinProjectionLayer
from ..distributions import UniversalGaussianDistribution, make_proba_distribution
from ..misc.distTools import get_mean_and_chol
metastable_baselines/ppo/ppo.py
+4 -4
View File
@@ -18,10 +18,10 @@ from stable_baselines3.common.vec_env import VecNormalize
from ..misc.distTools import new_dist_like, new_dist_like_from_sqrt
from ..projections.base_projection_layer import BaseProjectionLayer
from ..projections.frob_projection_layer import FrobeniusProjectionLayer
from ..projections.w2_projection_layer import WassersteinProjectionLayer
from ..projections.kl_projection_layer import KLProjectionLayer
from metastable_projections.projections.base_projection_layer import BaseProjectionLayer
from metastable_projections.projections.frob_projection_layer import FrobeniusProjectionLayer
from metastable_projections.projections.w2_projection_layer import WassersteinProjectionLayer
from metastable_projections.projections.kl_projection_layer import KLProjectionLayer
from ..misc.rollout_buffer import GaussianRolloutCollectorAuxclass