add vice learner with regularization techniques

serl-robot · Nov 2, 2023 · 5f30a74 · 5f30a74
1 parent 0df20d8
commit 5f30a74
Show file tree

Hide file tree

Showing 3 changed files with 474 additions and 3 deletions.
diff --git a/serl/agents/__init__.py b/serl/agents/__init__.py
@@ -1,11 +1,8 @@
 from serl.agents.bc.pixel_bc_learner import PixelBCLearner
 from serl.agents.bc.pixel_hybrid_bc_learner import PixelHybridBCLearner
 from serl.agents.sac.sac_learner import SACLearner
-from serl.agents.td3.td3_learner import TD3Learner
 from serl.agents.drq.drq_learner import DrQLearner
 from serl.agents.vice.vice_learner import VICELearner
-from serl.agents.vice.two_vice_learner import TwoVICELearner
-from serl.agents.vice.franka_vice_learner import FrankaVICELearner
 from serl.agents.drq.drq_classifier_reward_learner import DrQClassifierRewardLearner
 from serl.agents.ddpg.ddpg_learner import DDPGLearner
 from serl.agents.ddpg.pixel_ddpg_learner import PixelDDPGLearner
diff --git a/serl/agents/vice/__init__.py b/serl/agents/vice/__init__.py