Hugging Face's logo

YunzeLiu
/
fastwam-trex-egg-v2-roborag-p-5k

Robotics
fastwam
imitation-learning
vision-language-action
t-rex
bimanual-manipulation
Model card Files
xet
 Community
fastwam-trex-egg-v2-roborag-p-5k
File size: 2,442 Bytes
2029641
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "action_contract": {
    "frequency_hz": 30.0,
    "shape": [
      32,
      62
    ]
  },
  "baseline_repository": "YunzeLiu/fastwam-trex-egg-v2-baseline-5k",
  "checkpoint": {
    "path": "checkpoints/step_005000.pt",
    "role": "full_model",
    "sha256": "657b951e1e909a431bf90a2b6c0493dc1e29cae7b6c37ed401ed0501cd40b137",
    "size_bytes": 12051075733
  },
  "convergence_gate_passed": true,
  "dataset_path": "teleop_data/transfer_egg_v2/success",
  "dataset_revision": "d9057f25298097b5745bfd86af2727b544a4022e",
  "display_name": "roborag-p",
  "final_eval": {
    "action_l1": 0.016011182218790054,
    "action_l2": 0.001134719466790557,
    "event": "eval",
    "global_step": 5000,
    "psnr_dg": 40.04978942871094,
    "psnr_rd": 33.13883972167969,
    "psnr_rg": 28.606618881225586,
    "ssim_dg": 0.9873828291893005,
    "ssim_rd": 0.9363459348678589,
    "ssim_rg": 0.931090235710144,
    "time_unix": 1789595643.6236272,
    "val_loss": 0.31229186058044434,
    "video_path": "./runs/trex_egg_egodex_prefill_3cam448/trex62_egg_v2_roborag_p_from5k_20260916/eval/step_005000_rank_000.mp4"
  },
  "mode": "prefill",
  "real_robot_success_rate": null,
  "reference": {
    "context_tokens": {
      "dtype": "float16",
      "embedding_model_required_at_inference": false,
      "format": "safetensors",
      "mask_key": "context_mask",
      "path": "reference/context_tokens.safetensors",
      "precomputed": true,
      "role": "precomputed_context_tokens",
      "sha256": "7f29b6dd1d3c16baf7281b014c00a4091de72a8c964400ff2e31fa393055cbc6",
      "shape": [
        32,
        1024
      ],
      "size_bytes": 66176,
      "tensor_key": "context_tokens",
      "valid_tokens": 32
    },
    "feature_manifest_path": "reference/context_tokens_manifest.json",
    "video": {
      "path": "reference/reference.mp4",
      "role": "fixed_reference_video",
      "sha256": "6d9e3c9678e8f4427ff035fde8ecd85e8a8d094d1d1812c76b2a2ef52d2559b3",
      "size_bytes": 5169519
    }
  },
  "release_status": "validated_training_convergence",
  "repo_id": "YunzeLiu/fastwam-trex-egg-v2-roborag-p-5k",
  "schema_version": "fastwam-trex-egg-v2-final-v1",
  "source_commit": "ce76230cc4658ac8df6d3efe09c9f85e3f37a22a",
  "task_description": "Pick up the white egg from the orange egg cup with the right hand and place it in the purple cup.",
  "task_label": "T-Rex Egg v2",
  "training_done_marker": true,
  "training_step": 5000
}