Add verified full-suite LoopWAM v0 2/2 training checkpoint
Browse files- README.md +9 -1
- checkpoint_index.json +34 -1
- libero-all-suites/v0-video2-action2/README.md +7 -0
- libero-all-suites/v0-video2-action2/data_manifest.json +0 -0
- libero-all-suites/v0-video2-action2/dataset_stats.json +50 -0
- libero-all-suites/v0-video2-action2/export.json +33 -0
- libero-all-suites/v0-video2-action2/policy.pt +3 -0
- libero-all-suites/v0-video2-action2/training_manifest.json +0 -0
- libero-all-suites/v0-video2-action2/training_timing.json +49 -0
README.md
CHANGED
|
@@ -9,7 +9,7 @@ tags:
|
|
| 9 |
|
| 10 |
# LoopWAM final checkpoints
|
| 11 |
|
| 12 |
-
Final trained policy weights from the LoopWAM experiments. This repository contains **
|
| 13 |
|
| 14 |
## October 9 additions: all-loop KV
|
| 15 |
|
|
@@ -48,6 +48,14 @@ Normalization, training metadata, fairness checks and export hashes are included
|
|
| 48 |
|
| 49 |
The policy completed ten epochs on the four-suite full dataset with global batch 128. Evaluation results are not available for this checkpoint.
|
| 50 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
## Checkpoint index
|
| 52 |
|
| 53 |
| Directory | Model | Video/action loops | Parameters | Final evaluation |
|
|
|
|
| 9 |
|
| 10 |
# LoopWAM final checkpoints
|
| 11 |
|
| 12 |
+
Final trained policy weights from the LoopWAM experiments. This repository contains **16 checkpoints**, separated by training data scope. Both original and repeated 4/4 runs are retained. All production checkpoints are from completed ten-epoch runs with global batch 128 and training seed 42.
|
| 13 |
|
| 14 |
## October 9 additions: all-loop KV
|
| 15 |
|
|
|
|
| 48 |
|
| 49 |
The policy completed ten epochs on the four-suite full dataset with global batch 128. Evaluation results are not available for this checkpoint.
|
| 50 |
|
| 51 |
+
## October 10 addition: full-suite 2/2
|
| 52 |
+
|
| 53 |
+
| Directory | Model | Video/action loops | Parameters | Evaluation |
|
| 54 |
+
|---|---|---:|---:|---|
|
| 55 |
+
| [libero-all-suites/v0-video2-action2](libero-all-suites/v0-video2-action2) | v0 | 2/2 | 584,536,135 | Training only; evaluation not run |
|
| 56 |
+
|
| 57 |
+
The checkpoint completed ten epochs on the four-suite full dataset with global batch 128. Evaluation has not been run.
|
| 58 |
+
|
| 59 |
## Checkpoint index
|
| 60 |
|
| 61 |
| Directory | Model | Video/action loops | Parameters | Final evaluation |
|
checkpoint_index.json
CHANGED
|
@@ -399,7 +399,40 @@
|
|
| 399 |
"normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
|
| 400 |
"fairness_check_passed": true,
|
| 401 |
"created_utc": "2026-10-10T11:52:21.118949+00:00"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 402 |
}
|
| 403 |
],
|
| 404 |
-
"updated_utc": "2026-10-
|
| 405 |
}
|
|
|
|
| 399 |
"normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
|
| 400 |
"fairness_check_passed": true,
|
| 401 |
"created_utc": "2026-10-10T11:52:21.118949+00:00"
|
| 402 |
+
},
|
| 403 |
+
{
|
| 404 |
+
"id": "v0_full_2_2_recovery_job4824",
|
| 405 |
+
"repo_path": "libero-all-suites/v0-video2-action2",
|
| 406 |
+
"source_sha256": "a68000bf77d91b5c1f00bf2aae25b70a824bbe9601b8fb923a6a9d3681b04493",
|
| 407 |
+
"source_bytes": 7015569623,
|
| 408 |
+
"export_sha256": "5dd6ef39b7afb44a672750cebea57358eee43c7e998517e16f2c4f457c843600",
|
| 409 |
+
"export_bytes": 2338635000,
|
| 410 |
+
"format": "loopwam-s-v1",
|
| 411 |
+
"optimizer_included": false,
|
| 412 |
+
"step": 21700,
|
| 413 |
+
"version": "v0",
|
| 414 |
+
"video_loops": 2,
|
| 415 |
+
"action_loops": 2,
|
| 416 |
+
"action_kv_mode": "aligned",
|
| 417 |
+
"parameters": 584536135,
|
| 418 |
+
"dataset_scope": "full_libero",
|
| 419 |
+
"epochs": 10,
|
| 420 |
+
"global_batch": 128,
|
| 421 |
+
"training_seed": 42,
|
| 422 |
+
"evaluation_status": "not_run_training_only",
|
| 423 |
+
"success_rate": null,
|
| 424 |
+
"successes": null,
|
| 425 |
+
"episodes": 0,
|
| 426 |
+
"training_seconds": 37918.89883502503,
|
| 427 |
+
"initialization_mode": "resume_checkpoint",
|
| 428 |
+
"resumed_from_step": 4340,
|
| 429 |
+
"resume_checkpoint_sha256": "4e8a9e4175a095a5b682880e4d40892fc6fa971165681b7a5f5b02dbbe4372ae",
|
| 430 |
+
"tensor_equality_verified": true,
|
| 431 |
+
"source_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 432 |
+
"normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
|
| 433 |
+
"fairness_check_passed": true,
|
| 434 |
+
"created_utc": "2026-10-10T14:40:41.813293+00:00"
|
| 435 |
}
|
| 436 |
],
|
| 437 |
+
"updated_utc": "2026-10-10T14:40:42.538899+00:00"
|
| 438 |
}
|
libero-all-suites/v0-video2-action2/README.md
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# LoopWAM v0 2/2 — full LIBERO
|
| 2 |
+
|
| 3 |
+
Completed ten-epoch full-suite training checkpoint (Spatial, Object, Goal and LIBERO-10), seed 42, global batch 128, DDP on two H100 GPUs. This checkpoint has not been evaluated after training.
|
| 4 |
+
|
| 5 |
+
`policy.pt` is the native LoopWAM checkpoint with optimizer state removed. Its model tensors were checked for exact equality with the completed training checkpoint. It supports inference or fresh-optimizer fine-tuning, but not exact optimizer-state resume. Keep the accompanying `dataset_stats.json` matched to this checkpoint.
|
| 6 |
+
|
| 7 |
+
Training resumed from update 4,340 with saved optimizer and RNG state and completed at update 21,700. The training manifest, timing, data provenance, fairness checks and source/export hashes are included. See `export.json`.
|
libero-all-suites/v0-video2-action2/data_manifest.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
libero-all-suites/v0-video2-action2/dataset_stats.json
ADDED
|
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"action": {
|
| 3 |
+
"default": {
|
| 4 |
+
"global_max": [
|
| 5 |
+
0.9375,
|
| 6 |
+
0.9375,
|
| 7 |
+
0.9375,
|
| 8 |
+
0.3557142913341522,
|
| 9 |
+
0.375,
|
| 10 |
+
0.375,
|
| 11 |
+
1.0
|
| 12 |
+
],
|
| 13 |
+
"global_min": [
|
| 14 |
+
-0.9375,
|
| 15 |
+
-0.9375,
|
| 16 |
+
-0.9375,
|
| 17 |
+
-0.24214285612106323,
|
| 18 |
+
-0.375,
|
| 19 |
+
-0.3642857074737549,
|
| 20 |
+
0.0
|
| 21 |
+
]
|
| 22 |
+
}
|
| 23 |
+
},
|
| 24 |
+
"num_episodes": 1712,
|
| 25 |
+
"num_transition": 277713,
|
| 26 |
+
"state": {
|
| 27 |
+
"default": {
|
| 28 |
+
"global_max": [
|
| 29 |
+
0.2103137969970703,
|
| 30 |
+
0.3904264271259308,
|
| 31 |
+
1.472778081893921,
|
| 32 |
+
3.7248642444610596,
|
| 33 |
+
3.5618896484375,
|
| 34 |
+
1.3863215446472168,
|
| 35 |
+
0.04232141748070717,
|
| 36 |
+
0.0013126095291227102
|
| 37 |
+
],
|
| 38 |
+
"global_min": [
|
| 39 |
+
-0.48278069496154785,
|
| 40 |
+
-0.3309336006641388,
|
| 41 |
+
0.008128181099891663,
|
| 42 |
+
1.002794623374939,
|
| 43 |
+
-3.6312508583068848,
|
| 44 |
+
-1.842738389968872,
|
| 45 |
+
-0.005453015677630901,
|
| 46 |
+
-0.042015016078948975
|
| 47 |
+
]
|
| 48 |
+
}
|
| 49 |
+
}
|
| 50 |
+
}
|
libero-all-suites/v0-video2-action2/export.json
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"id": "v0_full_2_2_recovery_job4824",
|
| 3 |
+
"repo_path": "libero-all-suites/v0-video2-action2",
|
| 4 |
+
"source_sha256": "a68000bf77d91b5c1f00bf2aae25b70a824bbe9601b8fb923a6a9d3681b04493",
|
| 5 |
+
"source_bytes": 7015569623,
|
| 6 |
+
"export_sha256": "5dd6ef39b7afb44a672750cebea57358eee43c7e998517e16f2c4f457c843600",
|
| 7 |
+
"export_bytes": 2338635000,
|
| 8 |
+
"format": "loopwam-s-v1",
|
| 9 |
+
"optimizer_included": false,
|
| 10 |
+
"step": 21700,
|
| 11 |
+
"version": "v0",
|
| 12 |
+
"video_loops": 2,
|
| 13 |
+
"action_loops": 2,
|
| 14 |
+
"action_kv_mode": "aligned",
|
| 15 |
+
"parameters": 584536135,
|
| 16 |
+
"dataset_scope": "full_libero",
|
| 17 |
+
"epochs": 10,
|
| 18 |
+
"global_batch": 128,
|
| 19 |
+
"training_seed": 42,
|
| 20 |
+
"evaluation_status": "not_run_training_only",
|
| 21 |
+
"success_rate": null,
|
| 22 |
+
"successes": null,
|
| 23 |
+
"episodes": 0,
|
| 24 |
+
"training_seconds": 37918.89883502503,
|
| 25 |
+
"initialization_mode": "resume_checkpoint",
|
| 26 |
+
"resumed_from_step": 4340,
|
| 27 |
+
"resume_checkpoint_sha256": "4e8a9e4175a095a5b682880e4d40892fc6fa971165681b7a5f5b02dbbe4372ae",
|
| 28 |
+
"tensor_equality_verified": true,
|
| 29 |
+
"source_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 30 |
+
"normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
|
| 31 |
+
"fairness_check_passed": true,
|
| 32 |
+
"created_utc": "2026-10-10T14:40:41.813293+00:00"
|
| 33 |
+
}
|
libero-all-suites/v0-video2-action2/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5dd6ef39b7afb44a672750cebea57358eee43c7e998517e16f2c4f457c843600
|
| 3 |
+
size 2338635000
|
libero-all-suites/v0-video2-action2/training_manifest.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
libero-all-suites/v0-video2-action2/training_timing.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "complete",
|
| 3 |
+
"completed_updates": 21700,
|
| 4 |
+
"windows_seen": 2777130,
|
| 5 |
+
"planned_updates": 21700,
|
| 6 |
+
"elapsed_training_seconds": 37918.89883502503,
|
| 7 |
+
"checkpoint_seconds_total": 73.71588608506136,
|
| 8 |
+
"checkpoint_count": 8,
|
| 9 |
+
"measured_mean_update_seconds": 2.1760343434150142,
|
| 10 |
+
"estimated_total_training_hours": 13.11665145891828,
|
| 11 |
+
"estimate_note": "Projection from measured updates; excludes setup/checkpoint overhead",
|
| 12 |
+
"latest": {
|
| 13 |
+
"update": 21700,
|
| 14 |
+
"epoch": 10,
|
| 15 |
+
"windows": 81,
|
| 16 |
+
"windows_seen": 2777130,
|
| 17 |
+
"effective_passes": 10.0,
|
| 18 |
+
"seconds": 2.1505775451660156,
|
| 19 |
+
"lr": 1.0000000000000002e-06,
|
| 20 |
+
"grad_norm": 0.2209782600402832,
|
| 21 |
+
"branch_grad_after_clip": {
|
| 22 |
+
"video": 0.19537726044654846,
|
| 23 |
+
"action": 0.10207334160804749,
|
| 24 |
+
"proprio": 0.015496520325541496
|
| 25 |
+
},
|
| 26 |
+
"latent_cache_rank_local": {
|
| 27 |
+
"hits": 1110912,
|
| 28 |
+
"misses": 0,
|
| 29 |
+
"dummy_encodes": 0,
|
| 30 |
+
"encoder_samples": 0,
|
| 31 |
+
"batches": 69432,
|
| 32 |
+
"read_seconds": 1944.2269684574567,
|
| 33 |
+
"encode_seconds": 0.0,
|
| 34 |
+
"write_seconds": 0.0
|
| 35 |
+
},
|
| 36 |
+
"peak_allocated_gb": 68.81487274169922,
|
| 37 |
+
"peak_reserved_gb": 72.89490509033203,
|
| 38 |
+
"exit2/video_raw": 0.0545496881744008,
|
| 39 |
+
"exit2/action_raw": 0.049302006945197964,
|
| 40 |
+
"exit2/video": 0.06007812052597234,
|
| 41 |
+
"exit2/action": 0.023048068270271206,
|
| 42 |
+
"loss_video": 0.06007812052597234,
|
| 43 |
+
"loss_action": 0.023048068270271206,
|
| 44 |
+
"loop/1/video_state_rms": 1.371673866554543,
|
| 45 |
+
"loop/1/action_state_rms": 0.8940891218774113,
|
| 46 |
+
"loop/2/video_state_rms": 1.278758673020351,
|
| 47 |
+
"loop/2/action_state_rms": 0.9621150640793789
|
| 48 |
+
}
|
| 49 |
+
}
|