diff --git a/deploy_45dim_rl_gym/deploy_go1_onnx_mujoco_lab.py b/deploy_45dim_rl_gym/deploy_go1_onnx_mujoco_lab.py index 02188d0..a567866 100644 --- a/deploy_45dim_rl_gym/deploy_go1_onnx_mujoco_lab.py +++ b/deploy_45dim_rl_gym/deploy_go1_onnx_mujoco_lab.py @@ -43,7 +43,7 @@ from mujoco import viewer # ── path setup ── SCRIPT_DIR = Path(__file__).resolve().parent -DEFAULT_ONNX = str(SCRIPT_DIR / "policy_robotlab_6500.onnx") +DEFAULT_ONNX = str(SCRIPT_DIR / "policy_robotlab_15000.onnx") ROBOT_XML = str(SCRIPT_DIR / "go1.xml") TERRAINS_DIR = SCRIPT_DIR / "terrains" diff --git a/deploy_45dim_rl_gym/deploy_go1_rlgym_pro_sdk_lab_fastcpp.py b/deploy_45dim_rl_gym/deploy_go1_rlgym_pro_sdk_lab_fastcpp.py index 401b008..4a27de5 100644 --- a/deploy_45dim_rl_gym/deploy_go1_rlgym_pro_sdk_lab_fastcpp.py +++ b/deploy_45dim_rl_gym/deploy_go1_rlgym_pro_sdk_lab_fastcpp.py @@ -62,7 +62,7 @@ except ImportError as exc: "PYTHONPATH=/root/go1_pro_sdk python3 setup.py build_ext --inplace" ) from exc -DEFAULT_ONNX = HERE / "policy_robotlab_6500.onnx" +DEFAULT_ONNX = HERE / "policy_robotlab_15000.onnx" LOWCMD_BACKEND = "cpp_checked_servo12" SPORT_KILL_CMD = ( 'ssh pi@192.168.123.161 "sudo pkill -9 -f keep_sport_alive; ' diff --git a/deploy_45dim_rl_gym/policy_robotlab_15000.onnx b/deploy_45dim_rl_gym/policy_robotlab_15000.onnx new file mode 100644 index 0000000..27b592a Binary files /dev/null and b/deploy_45dim_rl_gym/policy_robotlab_15000.onnx differ