fastwam-trex-egg-v2-roborag-p-5k / checkpoint_manifest.json
YunzeLiu's picture
Add Egg v2 roborag-p configuration, features and provenance
2029641 verified
Raw
History Blame Contribute Delete
2.44 kB
{
"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
}