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/sac/sac.py
+4 -4
View File
@@ -16,10 +16,10 @@ from metastable_baselines.sac.policies import CnnPolicy, MlpPolicy, MultiInputPo
from ..misc.distTools import new_dist_like
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