feat: add Go1 training and MuJoCo deployment
This commit is contained in:
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user