Add verified full-suite LoopWAM v0 4/1 training checkpoint
Browse files- README.md +10 -2
- checkpoint_index.json +36 -1
- libero-all-suites/v0-video4-action1/README.md +7 -0
- libero-all-suites/v0-video4-action1/data_manifest.json +0 -0
- libero-all-suites/v0-video4-action1/dataset_stats.json +50 -0
- libero-all-suites/v0-video4-action1/export.json +35 -0
- libero-all-suites/v0-video4-action1/fairness.json +49 -0
- libero-all-suites/v0-video4-action1/policy.pt +3 -0
- libero-all-suites/v0-video4-action1/training_manifest.json +0 -0
- libero-all-suites/v0-video4-action1/training_timing.json +53 -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 |
|
|
@@ -25,7 +25,7 @@ equivalence. Each directory includes normalization, training provenance,
|
|
| 25 |
three evaluation summaries and a tensor-verified optimizer-free native policy.
|
| 26 |
The Long aligned4/1 and both original/repeat4/4 weights were hash-checked and
|
| 27 |
are unchanged. Their additional evaluation seeds do not create new weights.
|
| 28 |
-
The
|
| 29 |
|
| 30 |
Expanded seed42: concat404/500 versus aligned413/500 (paired p=0.4814). The
|
| 31 |
original three-seed decision remains inconclusive. This grid overlaps the
|
|
@@ -40,6 +40,14 @@ global batch 128, training seed 42). Training took 10h 58m 46s on two H100s.
|
|
| 40 |
**Not evaluated:** this run finished under the requested training-only workflow.
|
| 41 |
Normalization, training metadata, fairness checks and export hashes are included.
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
## Checkpoint index
|
| 44 |
|
| 45 |
| 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 **15 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 |
|
|
|
|
| 25 |
three evaluation summaries and a tensor-verified optimizer-free native policy.
|
| 26 |
The Long aligned4/1 and both original/repeat4/4 weights were hash-checked and
|
| 27 |
are unchanged. Their additional evaluation seeds do not create new weights.
|
| 28 |
+
The full-suite v0 4/1 training-only checkpoint is now included at `libero-all-suites/v0-video4-action1`. It has not been evaluated after training.
|
| 29 |
|
| 30 |
Expanded seed42: concat404/500 versus aligned413/500 (paired p=0.4814). The
|
| 31 |
original three-seed decision remains inconclusive. This grid overlaps the
|
|
|
|
| 40 |
**Not evaluated:** this run finished under the requested training-only workflow.
|
| 41 |
Normalization, training metadata, fairness checks and export hashes are included.
|
| 42 |
|
| 43 |
+
## October 10 addition: full-suite 4/1
|
| 44 |
+
|
| 45 |
+
| Directory | Model | Video/action loops | Parameters | Evaluation |
|
| 46 |
+
|---|---|---:|---:|---|
|
| 47 |
+
| [libero-all-suites/v0-video4-action1](libero-all-suites/v0-video4-action1) | v0 | 4/1 | 584,536,135 | Training only; evaluation not run |
|
| 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 |
|
checkpoint_index.json
CHANGED
|
@@ -364,7 +364,42 @@
|
|
| 364 |
"source_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 365 |
"compatible_code_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 366 |
"normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 367 |
}
|
| 368 |
],
|
| 369 |
-
"updated_utc": "2026-10-
|
| 370 |
}
|
|
|
|
| 364 |
"source_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 365 |
"compatible_code_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 366 |
"normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4"
|
| 367 |
+
},
|
| 368 |
+
{
|
| 369 |
+
"id": "v0_full_4_1_recovery_job4823",
|
| 370 |
+
"repo_path": "libero-all-suites/v0-video4-action1",
|
| 371 |
+
"training_path": "/mnt/data/vmo-ai-task/anhdh35/FastWAM/runs/loopwam_nt/storage_recovery_20261010/job4823_concat/full_41_resume/train",
|
| 372 |
+
"source_sha256": "e8c7882ea5149ba7c45c0692b0e5f813af88953beb0780bffd6978795f7f96ef",
|
| 373 |
+
"source_bytes": 7015569623,
|
| 374 |
+
"export_sha256": "d460ccea7148876d96e7b6969c97f698fb02119e12ab1f749872faf3d30c3a92",
|
| 375 |
+
"export_bytes": 2338635000,
|
| 376 |
+
"format": "loopwam-s-v1",
|
| 377 |
+
"optimizer_included": false,
|
| 378 |
+
"step": 21700,
|
| 379 |
+
"version": "v0",
|
| 380 |
+
"video_loops": 4,
|
| 381 |
+
"action_loops": 1,
|
| 382 |
+
"action_kv_mode": "aligned",
|
| 383 |
+
"parameters": 584536135,
|
| 384 |
+
"dataset_scope": "full_libero",
|
| 385 |
+
"epochs": 10,
|
| 386 |
+
"global_batch": 128,
|
| 387 |
+
"training_seed": 42,
|
| 388 |
+
"evaluation_status": "not_run_training_only",
|
| 389 |
+
"success_rate": null,
|
| 390 |
+
"successes": null,
|
| 391 |
+
"episodes": 0,
|
| 392 |
+
"recovery_segment_training_seconds": 22217.201475716894,
|
| 393 |
+
"initialization_mode": "resume_checkpoint",
|
| 394 |
+
"resumed_from_step": 15190,
|
| 395 |
+
"resume_checkpoint_sha256": "51939c93a4e3df01ac19d6847e360d2771343808fafd17c8ecfc5c09d74fb378",
|
| 396 |
+
"tensor_equality_verified": true,
|
| 397 |
+
"source_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 398 |
+
"compatible_code_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 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-10T11:52:21.130095+00:00"
|
| 405 |
}
|
libero-all-suites/v0-video4-action1/README.md
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# LoopWAM v0 4/1 — 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 is a training-only checkpoint; no post-training simulator evaluation has been run for this recovery checkpoint.
|
| 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 15,190 with the saved optimizer and RNG state; the recovery segment completed at update 21,700. The manifest, timing, data provenance, fairness check and source/export hashes are included. See `export.json`.
|
libero-all-suites/v0-video4-action1/data_manifest.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
libero-all-suites/v0-video4-action1/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-video4-action1/export.json
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"id": "v0_full_4_1_recovery_job4823",
|
| 3 |
+
"repo_path": "libero-all-suites/v0-video4-action1",
|
| 4 |
+
"training_path": "/mnt/data/vmo-ai-task/anhdh35/FastWAM/runs/loopwam_nt/storage_recovery_20261010/job4823_concat/full_41_resume/train",
|
| 5 |
+
"source_sha256": "e8c7882ea5149ba7c45c0692b0e5f813af88953beb0780bffd6978795f7f96ef",
|
| 6 |
+
"source_bytes": 7015569623,
|
| 7 |
+
"export_sha256": "d460ccea7148876d96e7b6969c97f698fb02119e12ab1f749872faf3d30c3a92",
|
| 8 |
+
"export_bytes": 2338635000,
|
| 9 |
+
"format": "loopwam-s-v1",
|
| 10 |
+
"optimizer_included": false,
|
| 11 |
+
"step": 21700,
|
| 12 |
+
"version": "v0",
|
| 13 |
+
"video_loops": 4,
|
| 14 |
+
"action_loops": 1,
|
| 15 |
+
"action_kv_mode": "aligned",
|
| 16 |
+
"parameters": 584536135,
|
| 17 |
+
"dataset_scope": "full_libero",
|
| 18 |
+
"epochs": 10,
|
| 19 |
+
"global_batch": 128,
|
| 20 |
+
"training_seed": 42,
|
| 21 |
+
"evaluation_status": "not_run_training_only",
|
| 22 |
+
"success_rate": null,
|
| 23 |
+
"successes": null,
|
| 24 |
+
"episodes": 0,
|
| 25 |
+
"recovery_segment_training_seconds": 22217.201475716894,
|
| 26 |
+
"initialization_mode": "resume_checkpoint",
|
| 27 |
+
"resumed_from_step": 15190,
|
| 28 |
+
"resume_checkpoint_sha256": "51939c93a4e3df01ac19d6847e360d2771343808fafd17c8ecfc5c09d74fb378",
|
| 29 |
+
"tensor_equality_verified": true,
|
| 30 |
+
"source_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 31 |
+
"compatible_code_revision": "3a89bfac5cfccad66bfebbe00014f7f739a78adc",
|
| 32 |
+
"normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
|
| 33 |
+
"fairness_check_passed": true,
|
| 34 |
+
"created_utc": "2026-10-10T11:52:21.118949+00:00"
|
| 35 |
+
}
|
libero-all-suites/v0-video4-action1/fairness.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"passed": true,
|
| 3 |
+
"compared_fields": [
|
| 4 |
+
"asset_sha256",
|
| 5 |
+
"seed",
|
| 6 |
+
"epochs",
|
| 7 |
+
"global_batch",
|
| 8 |
+
"train_windows",
|
| 9 |
+
"planned_updates",
|
| 10 |
+
"planned_windows",
|
| 11 |
+
"policy_dtype",
|
| 12 |
+
"optimizer_state_dtype",
|
| 13 |
+
"compute_dtype"
|
| 14 |
+
],
|
| 15 |
+
"data_fields": [
|
| 16 |
+
"action_offsets",
|
| 17 |
+
"available_episodes",
|
| 18 |
+
"available_windows",
|
| 19 |
+
"camera_order",
|
| 20 |
+
"codebase_version",
|
| 21 |
+
"complete_500_demo_set",
|
| 22 |
+
"content_files",
|
| 23 |
+
"coverage",
|
| 24 |
+
"dataset_dir",
|
| 25 |
+
"dataset_scope",
|
| 26 |
+
"episode_metadata_sha256",
|
| 27 |
+
"fps",
|
| 28 |
+
"info_sha256",
|
| 29 |
+
"metadata_files",
|
| 30 |
+
"normalization_scope",
|
| 31 |
+
"normalization_sha256",
|
| 32 |
+
"normalization_source",
|
| 33 |
+
"padding_policy",
|
| 34 |
+
"seed",
|
| 35 |
+
"split",
|
| 36 |
+
"split_algorithm",
|
| 37 |
+
"suites",
|
| 38 |
+
"task_counts",
|
| 39 |
+
"text_cache_dir",
|
| 40 |
+
"text_cache_files",
|
| 41 |
+
"text_encoder_provenance",
|
| 42 |
+
"train_episodes",
|
| 43 |
+
"train_windows",
|
| 44 |
+
"validation_episodes",
|
| 45 |
+
"validation_windows",
|
| 46 |
+
"video_offsets"
|
| 47 |
+
],
|
| 48 |
+
"accepted_initialization_change": true
|
| 49 |
+
}
|
libero-all-suites/v0-video4-action1/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d460ccea7148876d96e7b6969c97f698fb02119e12ab1f749872faf3d30c3a92
|
| 3 |
+
size 2338635000
|
libero-all-suites/v0-video4-action1/training_manifest.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
libero-all-suites/v0-video4-action1/training_timing.json
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "complete",
|
| 3 |
+
"completed_updates": 21700,
|
| 4 |
+
"windows_seen": 2777130,
|
| 5 |
+
"planned_updates": 21700,
|
| 6 |
+
"elapsed_training_seconds": 22217.201475716894,
|
| 7 |
+
"checkpoint_seconds_total": 26.169514174107462,
|
| 8 |
+
"checkpoint_count": 3,
|
| 9 |
+
"measured_mean_update_seconds": 3.4053813899455303,
|
| 10 |
+
"estimated_total_training_hours": 20.52688226717167,
|
| 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.8533496856689453,
|
| 19 |
+
"lr": 1.0000000000000002e-06,
|
| 20 |
+
"grad_norm": 0.2194993644952774,
|
| 21 |
+
"branch_grad_after_clip": {
|
| 22 |
+
"video": 0.2073885202407837,
|
| 23 |
+
"action": 0.07118500024080276,
|
| 24 |
+
"proprio": 0.010132784955203533
|
| 25 |
+
},
|
| 26 |
+
"latent_cache_rank_local": {
|
| 27 |
+
"hits": 416571,
|
| 28 |
+
"misses": 0,
|
| 29 |
+
"dummy_encodes": 21,
|
| 30 |
+
"encoder_samples": 21,
|
| 31 |
+
"batches": 52074,
|
| 32 |
+
"read_seconds": 848.248464504024,
|
| 33 |
+
"encode_seconds": 0.9370445790700614,
|
| 34 |
+
"write_seconds": 0.0
|
| 35 |
+
},
|
| 36 |
+
"peak_allocated_gb": 58.344120025634766,
|
| 37 |
+
"peak_reserved_gb": 65.10608673095703,
|
| 38 |
+
"exit4/video_raw": 0.08217585528338398,
|
| 39 |
+
"exit4/action_raw": 0.02396311730514338,
|
| 40 |
+
"exit4/video": 0.06038003497653537,
|
| 41 |
+
"exit4/action": 0.01664120179635507,
|
| 42 |
+
"loss_video": 0.06038003497653537,
|
| 43 |
+
"loss_action": 0.01664120179635507,
|
| 44 |
+
"loop/1/video_state_rms": 1.414136345003858,
|
| 45 |
+
"loop/1/action_state_rms": 0.7373409035765095,
|
| 46 |
+
"loop/2/video_state_rms": 1.5300074447820216,
|
| 47 |
+
"loop/2/action_state_rms": 0.7373409035765095,
|
| 48 |
+
"loop/3/video_state_rms": 1.647120346257716,
|
| 49 |
+
"loop/3/action_state_rms": 0.7373409035765095,
|
| 50 |
+
"loop/4/video_state_rms": 1.5038441787531347,
|
| 51 |
+
"loop/4/action_state_rms": 0.9207580943166473
|
| 52 |
+
}
|
| 53 |
+
}
|