{
  "model_dir": "/isaac-sim/.local/share/ov/data/groot/N1-2B",
  "arch": {
    "action_dim": 32,
    "action_horizon": 16,
    "diffusion_layers": 16,
    "diffusion_heads": 32,
    "input_emb_dim": 1536,
    "hidden_size": 1024,
    "backbone_name": "$GR00T_BACKBONE_PATH/eagle2_hg_model"
  },
  "tensor_inventory": {
    "total_tensors": 804,
    "action_head_M": 854.74,
    "backbone_B": 1.335,
    "other_M": 0.0,
    "action_head_samples": [
      {
        "name": "action_head.action_decoder.layer1.W",
        "shape": [
          32,
          1024,
          1024
        ],
        "dtype": "BF16",
        "norm": 150.78297424316406,
        "mean": 0.00023466398124583066,
        "std": 0.026031022891402245
      },
      {
        "name": "action_head.action_decoder.layer1.b",
        "shape": [
          32,
          1024
        ],
        "dtype": "BF16",
        "norm": 19.628002166748047,
        "mean": -0.046261146664619446,
        "std": 0.09806665033102036
      },
      {
        "name": "action_head.action_decoder.layer2.W",
        "shape": [
          32,
          1024,
          32
        ],
        "dtype": "BF16",
        "norm": 21.213775634765625,
        "mean": -2.0991639757994562e-05,
        "std": 0.020716369152069092
      },
      {
        "name": "action_head.action_decoder.layer2.b",
        "shape": [
          32,
          32
        ],
        "dtype": "BF16",
        "norm": 0.8890782594680786,
        "mean": -0.0008245159406214952,
        "std": 0.02777145430445671
      },
      {
        "name": "action_head.action_encoder.W1.W",
        "shape": [
          32,
          32,
          1536
        ],
        "dtype": "BF16",
        "norm": 86.19917297363281,
        "mean": -4.83581970911473e-07,
        "std": 0.06873174756765366
      }
    ]
  },
  "top_key_prefixes": [
    [
      "backbone.model",
      553
    ],
    [
      "action_head.model",
      232
    ],
    [
      "action_head.action_encoder",
      6
    ],
    [
      "action_head.action_decoder",
      4
    ],
    [
      "action_head.state_encoder",
      4
    ],
    [
      "action_head.decode_layer",
      2
    ],
    [
      "backbone.linear",
      2
    ],
    [
      "action_head.position_embedding",
      1
    ]
  ]
}