diff --git a/UPDATE.md b/UPDATE.md index 3c4c25d..0120b8d 100644 --- a/UPDATE.md +++ b/UPDATE.md @@ -1,3 +1,7 @@ +# 20260104 +## v0.1.3 +1. 修改moe-cts的torch script输出为`action, (weights, latent)`形式, cts输出为`action, (None, latent)` +2. 将cts,moe-cts的robogauge评测task都改为`go2_moe` # 20251231 ## v0.1.2 1. 加入PPO的Robogauge评估 diff --git a/rsl_rl/rsl_rl/runners/on_policy_runner_cts.py b/rsl_rl/rsl_rl/runners/on_policy_runner_cts.py index 38a8032..0c8de75 100644 --- a/rsl_rl/rsl_rl/runners/on_policy_runner_cts.py +++ b/rsl_rl/rsl_rl/runners/on_policy_runner_cts.py @@ -292,9 +292,7 @@ class OnPolicyRunnerCTS: jit_path = os.path.join(jit_dir, f'policy_jit_{it}.pt') export_policy_as_jit(self.alg.model, jit_dir, filename=f'policy_jit_{it}.pt') # upload to robogauge - task_name = 'go2' - if 'moe' in self.cfg["algorithm_class_name"].lower() or 'mcp' in self.cfg["algorithm_class_name"].lower(): - task_name = 'go2_moe' + task_name = 'go2_moe' # Both cts, moe-cts actor return a tuple `action, (latent, ...)` self.robogauge_client.submit_task( model_path=jit_path, step=it,