feat: add Go1 training and MuJoCo deployment

This commit is contained in:
youyuan.chen
2026-07-26 00:40:06 +08:00
parent 28b4516d22
commit 0d2afbf2bb
26 changed files with 2378 additions and 3 deletions

View File

@@ -12,6 +12,8 @@ Notes:
This script currently uses CONFIG_NAME = "go2.yaml" and does not expose CLI flags.
"""
from __future__ import annotations
import time
from pathlib import Path
@@ -95,9 +97,9 @@ def build_single_obs(features: dict[str, np.ndarray], layout: list[tuple[str, in
return np.concatenate([features[name] for name, _ in layout], axis=0).astype(np.float32, copy=False)
def main() -> None:
def main(config_name: str = CONFIG_NAME) -> None:
"""Run MuJoCo simulation and deploy the CTS policy in closed-loop control."""
cfg = load_config(CONFIG_NAME)
cfg = load_config(config_name)
layout = [
("ang_vel", 3),
("gravity", 3),