update cvpo config

liuzuxin · May 8, 2023 · 01c7de7 · 01c7de7
1 parent c989ed6
commit 01c7de7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/examples/mlp/train_cvpo_agent.py b/examples/mlp/train_cvpo_agent.py
@@ -56,7 +56,7 @@
  "SafetyHalfCheetahVelocityGymnasium-v1": Mujoco5MCfg,
  "SafetyHopperVelocityGymnasium-v1": Mujoco5MCfg,
  "SafetySwimmerVelocityGymnasium-v1": Mujoco5MCfg,
- "SafetyWalker2dVelocityGymnasium-v1": Mujoco10MCfg,
+ "SafetyWalker2dVelocityGymnasium-v1": MujocoBaseCfg,
  "SafetyAntVelocityGymnasium-v1": Mujoco10MCfg,
  "SafetyHumanoidVelocityGymnasium-v1": Mujoco20MCfg,
 }

diff --git a/fsrl/config/cvpo_cfg.py b/fsrl/config/cvpo_cfg.py
@@ -116,8 +116,6 @@ class Mujoco5MCfg(MujocoBaseCfg):
 @dataclass
 class Mujoco20MCfg(MujocoBaseCfg):
  epoch: int = 1000
- unbounded: bool = False
- gamma: float = 0.98
  sample_act_num: int = 64