Add libero-all-suites/init_variants/core_mean (four-suite v0 4/4, core_mean initialization, training only)

libero-all-suites/init_variants/core_mean/README.md

@@ -0,0 +1,21 @@
+# LoopWAM v0 (video 4 / action 4), initialization variant `core_mean` — all four LIBERO suites
+
+Same model, data and training as `libero-all-suites/v0-video4-action4`; only the student initialization differs:
+core block i initialized as the element-wise mean of Wan blocks 3+i, 9+i, 15+i, 21+i (after the canonical FFN pruning / action resizing). Prelude (Wan blocks 0-2) and coda (Wan blocks 27-29) are as in v0.
+
+Ten-epoch full-suite training (Spatial, Object, Goal and LIBERO-10; 1,712 demonstrations, 277,713 windows), training seed
+42, global batch 128 (2 GPUs x microbatch 16 x accumulation
+4), 21,700 updates, AdamW (0.9, 0.95), weight decay 0.01, LR 1e-4 cosine with 5% warm-up,
+clip 1.0, FP32 master weights + BF16 compute, 584,536,135 policy parameters. Data, normalization
+(`d5ad351eb327...`), VAE, schedule and precision equal the v0 release; `fairness.json` lists the
+checked fields and `init_variant_manifest.json` records the initialization artifact (SHA-256
+`506bcecfeac2...`) and its per-block source plan. Training took 12.8 h
+on 2x NVIDIA B200. Training ran in one segment.
+
+**Not evaluated:** this is a training-only checkpoint; no simulator evaluation has been run and no success rate is claimed.
+
+`policy.pt` is the native `loopwam-s-v1` checkpoint with the Adam optimizer state removed; every model tensor was
+checked for exact equality with the final training checkpoint. Use it for inference or fresh-optimizer fine-tuning,
+not exact optimizer-state resume. Keep `dataset_stats.json` matched to it. Load with the LoopWAM code (revision
+`4ebf38a3c65aa5cc9d6c07d94c52f91c51551b57` or a compatible later one): `create_loopwam(checkpoint_path=..., vae_path=...)`. Wan VAE,
+text embeddings and observation preprocessing are external.

libero-all-suites/init_variants/core_mean/dataset_stats.json

@@ -0,0 +1,50 @@
+{
+  "action": {
+    "default": {
+      "global_max": [
+        0.9375,
+        0.9375,
+        0.9375,
+        0.3557142913341522,
+        0.375,
+        0.375,
+        1.0
+      ],
+      "global_min": [
+        -0.9375,
+        -0.9375,
+        -0.9375,
+        -0.24214285612106323,
+        -0.375,
+        -0.3642857074737549,
+        0.0
+      ]
+    }
+  },
+  "num_episodes": 1712,
+  "num_transition": 277713,
+  "state": {
+    "default": {
+      "global_max": [
+        0.2103137969970703,
+        0.3904264271259308,
+        1.472778081893921,
+        3.7248642444610596,
+        3.5618896484375,
+        1.3863215446472168,
+        0.04232141748070717,
+        0.0013126095291227102
+      ],
+      "global_min": [
+        -0.48278069496154785,
+        -0.3309336006641388,
+        0.008128181099891663,
+        1.002794623374939,
+        -3.6312508583068848,
+        -1.842738389968872,
+        -0.005453015677630901,
+        -0.042015016078948975
+      ]
+    }
+  }
+}

libero-all-suites/init_variants/core_mean/export.json

@@ -0,0 +1,35 @@
+{
+  "id": "v0_core_mean_full_libero_hpg",
+  "repo_path": "libero-all-suites/init_variants/core_mean",
+  "training_path": "/blue/yu954774.ucf/an221229.ucf/checkpoints/LoopWAM/runs/v0_init_variants/core_mean/train",
+  "source_sha256": "ba21e88007fb570577d90bd8c961a7e8eeb671fdcf35c9faf8d95e3bf466e66b",
+  "source_bytes": 7015676119,
+  "export_sha256": "7613e262c47bd848af49af6797fab37f0d6a199032312a08cafabaa36d004664",
+  "export_bytes": 2338749835,
+  "format": "loopwam-s-v1",
+  "optimizer_included": false,
+  "step": 21700,
+  "version": "v0",
+  "video_loops": 4,
+  "action_loops": 4,
+  "parameters": 584536135,
+  "init_variant": "core_mean",
+  "initialization_sha256": "506bcecfeac2f2cf2eadfccffdd68060bd5f5ccc24f85bf2333a959b0001bef0",
+  "initialization_mode": "wan_init_variant_core_mean_fresh_optimizer",
+  "dataset_scope": "full_libero",
+  "epochs": 10,
+  "global_batch": 128,
+  "training_seed": 42,
+  "evaluation_status": "not_run_training_only",
+  "success_rate": null,
+  "successes": null,
+  "episodes": 0,
+  "training_seconds": 45977.33255086746,
+  "tensor_equality_verified": true,
+  "source_revision": "4ebf38a3c65aa5cc9d6c07d94c52f91c51551b57",
+  "normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
+  "fairness_check_passed": true,
+  "gpu": "NVIDIA B200",
+  "world_size": 2,
+  "created_utc": "2026-10-27T23:03:15.755642+00:00"
+}

libero-all-suites/init_variants/core_mean/fairness.json

@@ -0,0 +1,66 @@
+{
+  "passed": true,
+  "problems": [],
+  "version": "v0",
+  "init_variant": "core_mean",
+  "initialization_sha256": "506bcecfeac2f2cf2eadfccffdd68060bd5f5ccc24f85bf2333a959b0001bef0",
+  "reference": "/blue/yu954774.ucf/an221229.ucf/checkpoints/ChronoLoop/parent/libero-all-suites/v0-video4-action4",
+  "compared_training_fields": [
+    "seed",
+    "epochs",
+    "global_batch",
+    "train_windows",
+    "planned_updates",
+    "planned_windows",
+    "policy_dtype",
+    "optimizer_state_dtype",
+    "compute_dtype"
+  ],
+  "data_fields": [
+    "action_offsets",
+    "available_episodes",
+    "available_windows",
+    "camera_order",
+    "codebase_version",
+    "complete_500_demo_set",
+    "content_files",
+    "coverage",
+    "dataset_scope",
+    "episode_metadata_sha256",
+    "fps",
+    "info_sha256",
+    "metadata_files",
+    "normalization_scope",
+    "normalization_sha256",
+    "normalization_source",
+    "padding_policy",
+    "seed",
+    "split",
+    "split_algorithm",
+    "suites",
+    "task_counts",
+    "text_cache_files",
+    "text_encoder_provenance",
+    "train_episodes",
+    "train_windows",
+    "validation_episodes",
+    "validation_windows",
+    "video_offsets"
+  ],
+  "ignored_path_fields": [
+    "dataset_dir",
+    "normalization_path",
+    "text_cache_dir"
+  ],
+  "world_size": 2,
+  "microbatch": 16,
+  "gradient_accumulation": 4,
+  "gpu": "NVIDIA B200",
+  "git_revision": "4ebf38a3c65aa5cc9d6c07d94c52f91c51551b57",
+  "resumed": false,
+  "resume_manifests": [],
+  "normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
+  "final_epoch_loss_video": 0.0528129199272736,
+  "final_epoch_loss_action": 0.016135175192377133,
+  "training_hours": 12.77148126412985
+}

libero-all-suites/init_variants/core_mean/init_variant_manifest.json

@@ -0,0 +1,106 @@
+{
+  "variant": "core_mean",
+  "artifact": "checkpoints/LoopWAM/init_variants/core_mean.pt",
+  "artifact_sha256": "506bcecfeac2f2cf2eadfccffdd68060bd5f5ccc24f85bf2333a959b0001bef0",
+  "canonical_artifact": "checkpoints/LoopWAM/wan21_compact_donors.pt",
+  "canonical_artifact_sha256": "a3bf2097bafddac278148c1cb0321163f68cfe2b954e206436da0e3f0bd3c535",
+  "student_block_plan": [
+    [
+      "wan",
+      0
+    ],
+    [
+      "wan",
+      1
+    ],
+    [
+      "wan",
+      2
+    ],
+    [
+      "mean",
+      [
+        3,
+        9,
+        15,
+        21
+      ]
+    ],
+    [
+      "mean",
+      [
+        4,
+        10,
+        16,
+        22
+      ]
+    ],
+    [
+      "mean",
+      [
+        5,
+        11,
+        17,
+        23
+      ]
+    ],
+    [
+      "mean",
+      [
+        6,
+        12,
+        18,
+        24
+      ]
+    ],
+    [
+      "mean",
+      [
+        7,
+        13,
+        19,
+        25
+      ]
+    ],
+    [
+      "mean",
+      [
+        8,
+        14,
+        20,
+        26
+      ]
+    ],
+    [
+      "wan",
+      27
+    ],
+    [
+      "wan",
+      28
+    ],
+    [
+      "wan",
+      29
+    ]
+  ],
+  "donor_indices": null,
+  "ffn_rule": "canonical_linspace_6144",
+  "variant_seed": 42,
+  "policy_parameters": 584536135,
+  "checks": {
+    "tensors": 677,
+    "copied": 353,
+    "mean": 324,
+    "random": 0,
+    "magnitude_ffn": 0
+  },
+  "core_slots": [
+    3,
+    4,
+    5,
+    6,
+    7,
+    8
+  ]
+}

libero-all-suites/init_variants/core_mean/policy.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7613e262c47bd848af49af6797fab37f0d6a199032312a08cafabaa36d004664
+size 2338749835

libero-all-suites/init_variants/core_mean/training_timing.json

@@ -0,0 +1,53 @@
+{
+  "status": "complete",
+  "completed_updates": 21700,
+  "windows_seen": 2777130,
+  "planned_updates": 21700,
+  "elapsed_training_seconds": 45977.33255086746,
+  "checkpoint_seconds_total": 86.63813340850174,
+  "checkpoint_count": 10,
+  "measured_mean_update_seconds": 2.113643960835412,
+  "estimated_total_training_hours": 12.740576097257899,
+  "estimate_note": "Projection from measured updates; excludes setup/checkpoint overhead",
+  "latest": {
+    "update": 21700,
+    "epoch": 10,
+    "windows": 81,
+    "windows_seen": 2777130,
+    "effective_passes": 10.0,
+    "seconds": 1.9898278713226318,
+    "lr": 1.0000000000000002e-06,
+    "grad_norm": 0.259105384349823,
+    "branch_grad_after_clip": {
+      "video": 0.23641441762447357,
+      "action": 0.10414916276931763,
+      "proprio": 0.019919270649552345
+    },
+    "latent_cache_rank_local": {
+      "hits": 1388640,
+      "misses": 0,
+      "dummy_encodes": 0,
+      "encoder_samples": 0,
+      "batches": 86790,
+      "read_seconds": 1524.9051061663777,
+      "encode_seconds": 0.0,
+      "write_seconds": 0.0
+    },
+    "peak_allocated_gb": 104.78271484375,
+    "peak_reserved_gb": 127.83818817138672,
+    "exit4/video_raw": 0.04980073740452896,
+    "exit4/action_raw": 0.040114499904491285,
+    "exit4/video": 0.046064641740587026,
+    "exit4/action": 0.017163044140662675,
+    "loss_video": 0.046064641740587026,
+    "loss_action": 0.017163044140662675,
+    "loop/1/video_state_rms": 0.9373847113715278,
+    "loop/1/action_state_rms": 0.8380150500638985,
+    "loop/2/video_state_rms": 1.0897996690538194,
+    "loop/2/action_state_rms": 0.902228272991416,
+    "loop/3/video_state_rms": 1.251085540394724,
+    "loop/3/action_state_rms": 0.9654291411976755,
+    "loop/4/video_state_rms": 1.343444730028694,
+    "loop/4/action_state_rms": 1.02816621756848
+  }
+}