[misc]: Pika H3 Flex Ref2VA QAD run kit, pinned calibration table, final recipe config

examples/train/calibration/pika_h3_flex/amax-ref2va-balanced.json

@@ -0,0 +1,102 @@
+{
+ "b0.ff.fc_in": 5.75,
+ "b0.ff.fc_out": 1792.0,
+ "b1.ff.fc_in": 8.9375,
+ "b1.ff.fc_out": 2704.0,
+ "b2.ff.fc_in": 4.5625,
+ "b2.ff.fc_out": 2096.0,
+ "b3.ff.fc_in": 6.875,
+ "b3.ff.fc_out": 2384.0,
+ "b4.ff.fc_in": 11.125,
+ "b4.ff.fc_out": 2720.0,
+ "b5.ff.fc_in": 18.549999999999997,
+ "b5.ff.fc_out": 8225.280000000006,
+ "b6.ff.fc_in": 33.0,
+ "b6.ff.fc_out": 4352.0,
+ "b7.ff.fc_in": 18.75,
+ "b7.ff.fc_out": 4608.0,
+ "b8.ff.fc_in": 13.375,
+ "b8.ff.fc_out": 2320.0,
+ "b9.ff.fc_in": 13.212499999999999,
+ "b9.ff.fc_out": 2890.880000000001,
+ "b10.ff.fc_in": 23.875,
+ "b10.ff.fc_out": 4320.0,
+ "b11.ff.fc_in": 13.737499999999999,
+ "b11.ff.fc_out": 6227.2,
+ "b12.ff.fc_in": 14.874999999999998,
+ "b12.ff.fc_out": 1632.0,
+ "b13.ff.fc_in": 15.049999999999999,
+ "b13.ff.fc_out": 22144.0,
+ "b14.ff.fc_in": 7.46875,
+ "b14.ff.fc_out": 1552.0,
+ "b15.ff.fc_in": 9.4375,
+ "b15.ff.fc_out": 1444.8,
+ "b16.ff.fc_in": 11.272500000000008,
+ "b16.ff.fc_out": 3216.0,
+ "b17.ff.fc_in": 30.5,
+ "b17.ff.fc_out": 3760.0,
+ "b18.ff.fc_in": 11.4375,
+ "b18.ff.fc_out": 14656.0,
+ "b19.ff.fc_in": 12.5,
+ "b19.ff.fc_out": 4144.0,
+ "b20.ff.fc_in": 15.5,
+ "b20.ff.fc_out": 4544.0,
+ "b21.ff.fc_in": 10.149999999999999,
+ "b21.ff.fc_out": 9920.0,
+ "b22.ff.fc_in": 14.524999999999999,
+ "b22.ff.fc_out": 6656.0,
+ "b23.ff.fc_in": 18.75,
+ "b23.ff.fc_out": 8288.0,
+ "b24.ff.fc_in": 12.987500000000004,
+ "b24.ff.fc_out": 11520.0,
+ "b25.ff.fc_in": 14.625,
+ "b25.ff.fc_out": 9963.520000000004,
+ "b26.ff.fc_in": 10.0625,
+ "b26.ff.fc_out": 9390.080000000016,
+ "b27.ff.fc_in": 11.5625,
+ "b27.ff.fc_out": 6784.0,
+ "b28.ff.fc_in": 10.587499999999999,
+ "b28.ff.fc_out": 9900.8,
+ "b29.ff.fc_in": 8.44375,
+ "b29.ff.fc_out": 11991.040000000008,
+ "b30.ff.fc_in": 10.25,
+ "b30.ff.fc_out": 15411.199999999999,
+ "b31.ff.fc_in": 11.355000000000004,
+ "b31.ff.fc_out": 8576.0,
+ "b32.ff.fc_in": 12.9375,
+ "b32.ff.fc_out": 12928.0,
+ "b33.ff.fc_in": 11.125,
+ "b33.ff.fc_out": 9907.200000000004,
+ "b34.ff.fc_in": 10.230000000000004,
+ "b34.ff.fc_out": 14656.0,
+ "b35.ff.fc_in": 13.75,
+ "b35.ff.fc_out": 15360.0,
+ "b36.ff.fc_in": 14.417500000000004,
+ "b36.ff.fc_out": 29568.0,
+ "b37.ff.fc_in": 11.4375,
+ "b37.ff.fc_out": 12759.040000000008,
+ "b38.ff.fc_in": 19.085000000000008,
+ "b38.ff.fc_out": 30848.0,
+ "b39.ff.fc_in": 8.44375,
+ "b39.ff.fc_out": 290816.0,
+ "b40.ff.fc_in": 12.272500000000008,
+ "b40.ff.fc_out": 20787.199999999997,
+ "b41.ff.fc_in": 10.8125,
+ "b41.ff.fc_out": 18887.68000000014,
+ "b42.ff.fc_in": 11.06875,
+ "b42.ff.fc_out": 11264.0,
+ "b43.ff.fc_in": 16.275,
+ "b43.ff.fc_out": 25728.0,
+ "b44.ff.fc_in": 36.75,
+ "b44.ff.fc_out": 54476.799999999996,
+ "b45.ff.fc_in": 45.51000000000005,
+ "b45.ff.fc_out": 118272.0,
+ "b46.ff.fc_in": 40.20000000000002,
+ "b46.ff.fc_out": 20070.399999999998,
+ "b47.ff.fc_in": 54.0,
+ "b47.ff.fc_out": 31232.0,
+ "b48.ff.fc_in": 113.5,
+ "b48.ff.fc_out": 40960.0,
+ "b49.ff.fc_in": 128.0,
+ "b49.ff.fc_out": 167731.19999999998
+}

examples/train/configs/knowledge_distillation/pika_h3_flex/ref2va_nvfp4_qad.yaml

@@ -26,8 +26,8 @@ models:
       enabled: true
       linears:
         enabled: true
-        amax_json: examples/train/calibration/pika_h3_flex/amax-ref2va.json
-        amax_sha256: null
+        amax_json: examples/train/calibration/pika_h3_flex/amax-ref2va-balanced.json
+        amax_sha256: ad60b45fea00906c55e370a0dec82638eb413d6b68537d1ceb38fc52da90b373
         # Attention linears use the unit activation scale; only the FFN
         # linears read their calibrated amax from this table.
         attention: true
@@ -37,6 +37,8 @@ models:
         enabled: false
     data:
       manifests:
+      # Flex-sized rows (build_flex_ref2va_rows.py) first: they replace v4 rows with the same id.
+      - data/flex-ref2va/MANIFEST.jsonl
       - data/omniref/train-manifest.jsonl
       cases:
       - first_frame
@@ -44,13 +46,16 @@ models:
       - storyboard
       - continue_scene
       - continue_shot
-      - full_scene_audio_image_reference
+      - audio_reference
+      - mixed_image_video
+      - many_images
+      - multi_video
       per_group:
-        480p: 400
-        768p: 50
+        480p: 160
+        768p: 40
       seed: 20261007
-      max_frames: 243
-      eval_manifest: data/omniref/eval-manifest.json
+      max_frames: 345
+      eval_manifest: data/flex-ref2va/eval-manifest.json
       eval_resolutions:
       - 480p
       - 768p