anhdao69 commited on
Commit
969a358
·
verified ·
1 Parent(s): 9045349

Add verified full-suite LoopWAM v0 4/1 training checkpoint

Browse files
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 **14 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,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 new full-suite4/1 training is ongoing; no intermediate checkpoint is uploaded.
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-10T02:41:05.842410+00:00"
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
+ }