Hugging Face's logo

nvidia
/
GEAR-SONIC

Reinforcement Learning
ONNX
PyTorch
robotics
humanoid
whole-body-control
motion-tracking
teleoperation
isaac-lab
Model card Files
xet
 Community
GEAR-SONIC
File size: 2,428 Bytes
bfa2404
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
{
  "schema_version": 1,
  "model_id": "nvidia/GEAR-SONIC",
  "model_name": "GEAR-SONIC",
  "embodiment": "Unitree G1",
  "control_frequency_hz": 50,
  "token_dimension": 64,
  "variants": {
    "default": {
      "display_name": "Default SONIC",
      "description": "Original general-purpose SONIC whole-body controller.",
      "smpl_reference": {
        "future_frames": 10,
        "frame_interval_ms": 20,
        "lookahead_ms": 200
      },
      "deployment_files": [
        "model_encoder.onnx",
        "model_decoder.onnx",
        "observation_config.yaml"
      ],
      "training_files": [
        "sonic_release/last.pt",
        "sonic_release/config.yaml"
      ]
    },
    "low_latency": {
      "display_name": "Low-latency teleoperation",
      "description": "Whole-body teleoperation controller with a shorter SMPL reference horizon.",
      "smpl_reference": {
        "future_frames": 4,
        "frame_interval_ms": 20,
        "lookahead_ms": 80
      },
      "deployment_files": [
        "low_latency/model_encoder.onnx",
        "low_latency/model_decoder.onnx",
        "low_latency/observation_config.yaml"
      ],
      "training_files": [
        "low_latency/last.pt",
        "low_latency/config.yaml",
        "low_latency/model_config.yaml"
      ]
    },
    "sonic_v1_1": {
      "display_name": "SONIC v1.1",
      "description": "Whole-body teleoperation controller with robot-heading-normalized targets and wrist-pose augmentation.",
      "smpl_reference": {
        "future_frames": 10,
        "frame_interval_ms": 20,
        "lookahead_ms": 200
      },
      "deployment_files": [
        "sonic_v1_1/model_encoder.onnx",
        "sonic_v1_1/model_decoder.onnx",
        "sonic_v1_1/observation_config.yaml"
      ],
      "training_files": [
        "sonic_v1_1/last.pt",
        "sonic_v1_1/config.yaml",
        "sonic_v1_1/model_config.yaml"
      ]
    }
  },
  "shared_files": {
    "planner": "planner_sonic.onnx",
    "sample_data_prefix": "sample_data/",
    "smpl_archive_parts": [
      "bones_seed_smpl/bones_seed_smpl.tar.part_aa",
      "bones_seed_smpl/bones_seed_smpl.tar.part_ab",
      "bones_seed_smpl/bones_seed_smpl.tar.part_ac",
      "bones_seed_smpl/bones_seed_smpl.tar.part_ad",
      "bones_seed_smpl/bones_seed_smpl.tar.part_ae",
      "bones_seed_smpl/bones_seed_smpl.tar.part_af",
      "bones_seed_smpl/bones_seed_smpl.tar.part_ag"
    ]
  }
}