v0.1.3; change robogauge taskname

This commit is contained in:
wty-yy
2026-01-04 15:51:43 +08:00
parent c2519bb141
commit 319853d0f3
2 changed files with 5 additions and 3 deletions

View File

@@ -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评估

View File

@@ -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,