Upload folder using huggingface_hub

p4rel-mae-15m-d70-s0/config.yaml

@@ -0,0 +1,112 @@
+model:
+  embed_dim: 384
+  depth: 8
+  num_heads: 6
+  mlp_ratio: 4
+  target_frames: 1024
+  mel_bins: 64
+  patch_time: 16
+  patch_freq: 16
+  decoder_layers: 6
+  decoder_groups: 16
+  decoder_kernel: 3
+objective:
+  kind: mae
+  mae_norm_pix: true
+  num_classes: 527
+  clone_batch: 1
+  mask_prob: 0.8
+  mask_length: 5
+  mask_prob_adjust: 0.07
+  average_top_k_layers: 0
+  utterance_weight: 1.0
+  normalize_utterance_target: false
+  ema_decay: 0.9998
+  ema_end_decay: 0.99999
+  ema_anneal_steps: null
+data:
+  data_dir: /workspace/data/asmel_flat
+  train_split: train
+  val_split: balanced_train
+  revision: ''
+  batch_size: 48
+  num_workers: 12
+  prefetch_factor: 4
+  pin_memory: true
+  persistent_workers: true
+  drop_last: true
+  seed: 0
+  max_shards: 0
+mixture:
+  cluster_index: /workspace/data/asmel_flat/train_cluster_index.npy
+  distributions: null
+  dist_id: null
+  weights:
+  - 0.004603515625000001
+  - 0.05805078125000001
+  - 0.0
+  - 0.25281054687500004
+  - 0.06202539062500001
+  - 0.0
+  - 0.16547265625000002
+  - 0.0
+  - 0.0021972656250000004
+  - 0.0
+  - 0.007425781250000001
+  - 0.016421875000000002
+  - 0.0008886718750000001
+  - 0.00041796875000000005
+  - 0.14711523437500001
+  - 0.0
+  - 0.24878515625000003
+  - 0.0027207031250000005
+  - 0.026671875000000005
+  - 0.004392578125000001
+  repetition_cap: 1000.0
+  distinct_fraction: null
+optim:
+  lr: 0.0003
+  betas:
+  - 0.9
+  - 0.95
+  eps: 1.0e-08
+  weight_decay: 0.05
+  clip_norm: 4.0
+  warmup_frac: 0.1
+  min_lr_scale: 0.002
+loop:
+  budget_clips: 1920000
+  schedule_clips: null
+  gradient_accumulation_steps: 1
+  precision: bfloat16
+  compile: true
+  compile_mode: default
+  log_every_n_steps: 20
+  heartbeat_every_n_steps: 50
+  max_steps: null
+eval:
+  enabled: true
+  every_n_steps: 10000
+  num_clips: 2048
+  mask_seed: 12345
+  batch_size: 48
+checkpoint:
+  save_every_n_steps: 40000
+  export_every_n_steps: 40000
+  keep_last_n_exports: 0
+tracking:
+  campaign_id: phase4-reliability
+  trial_id: p4rel-mae-15m-d70-s0
+  wandb_project: mixture-transfer
+  wandb_entity: null
+  wandb_mode: online
+  wandb_run_id: null
+  hub_repo_id: null
+  hub_private: false
+  hub_upload: false
+runtime:
+  seed: 0
+  data_loader_seed: null
+  objective_seed: null
+  deterministic: false
+  allow_tf32: true

p4rel-mae-15m-d70-s0/exports/step_00040000/config.yaml

@@ -0,0 +1,112 @@
+model:
+  embed_dim: 384
+  depth: 8
+  num_heads: 6
+  mlp_ratio: 4
+  target_frames: 1024
+  mel_bins: 64
+  patch_time: 16
+  patch_freq: 16
+  decoder_layers: 6
+  decoder_groups: 16
+  decoder_kernel: 3
+objective:
+  kind: mae
+  mae_norm_pix: true
+  num_classes: 527
+  clone_batch: 1
+  mask_prob: 0.8
+  mask_length: 5
+  mask_prob_adjust: 0.07
+  average_top_k_layers: 0
+  utterance_weight: 1.0
+  normalize_utterance_target: false
+  ema_decay: 0.9998
+  ema_end_decay: 0.99999
+  ema_anneal_steps: null
+data:
+  data_dir: /workspace/data/asmel_flat
+  train_split: train
+  val_split: balanced_train
+  revision: ''
+  batch_size: 48
+  num_workers: 12
+  prefetch_factor: 4
+  pin_memory: true
+  persistent_workers: true
+  drop_last: true
+  seed: 0
+  max_shards: 0
+mixture:
+  cluster_index: /workspace/data/asmel_flat/train_cluster_index.npy
+  distributions: null
+  dist_id: null
+  weights:
+  - 0.004603515625000001
+  - 0.05805078125000001
+  - 0.0
+  - 0.25281054687500004
+  - 0.06202539062500001
+  - 0.0
+  - 0.16547265625000002
+  - 0.0
+  - 0.0021972656250000004
+  - 0.0
+  - 0.007425781250000001
+  - 0.016421875000000002
+  - 0.0008886718750000001
+  - 0.00041796875000000005
+  - 0.14711523437500001
+  - 0.0
+  - 0.24878515625000003
+  - 0.0027207031250000005
+  - 0.026671875000000005
+  - 0.004392578125000001
+  repetition_cap: 1000.0
+  distinct_fraction: null
+optim:
+  lr: 0.0003
+  betas:
+  - 0.9
+  - 0.95
+  eps: 1.0e-08
+  weight_decay: 0.05
+  clip_norm: 4.0
+  warmup_frac: 0.1
+  min_lr_scale: 0.002
+loop:
+  budget_clips: 1920000
+  schedule_clips: null
+  gradient_accumulation_steps: 1
+  precision: bfloat16
+  compile: true
+  compile_mode: default
+  log_every_n_steps: 20
+  heartbeat_every_n_steps: 50
+  max_steps: null
+eval:
+  enabled: true
+  every_n_steps: 10000
+  num_clips: 2048
+  mask_seed: 12345
+  batch_size: 48
+checkpoint:
+  save_every_n_steps: 40000
+  export_every_n_steps: 40000
+  keep_last_n_exports: 0
+tracking:
+  campaign_id: phase4-reliability
+  trial_id: p4rel-mae-15m-d70-s0
+  wandb_project: mixture-transfer
+  wandb_entity: null
+  wandb_mode: online
+  wandb_run_id: null
+  hub_repo_id: null
+  hub_private: false
+  hub_upload: false
+runtime:
+  seed: 0
+  data_loader_seed: null
+  objective_seed: null
+  deterministic: false
+  allow_tf32: true

p4rel-mae-15m-d70-s0/exports/step_00040000/fsd50k_lora_1ep_s0.json

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-15m-d70-s0/exports/step_00040000",
+  "fingerprint": "06edcd5f801cd692",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d70-s0",
+  "target": "fsd50k",
+  "num_classes": 200,
+  "lora_layers_wrapped": 32,
+  "params": {
+    "trainable": 876488,
+    "total": 15158984,
+    "fraction": 0.05781970612278501
+  },
+  "steps_per_epoch": 989,
+  "total_steps": 989,
+  "train_clips": 36796,
+  "train_segments": 47487,
+  "val_clips": 10231,
+  "val_segments": 15513,
+  "train_split": "train",
+  "eval_split": "test",
+  "config": {
+    "rank": 16,
+    "alpha": 32.0,
+    "targets": [
+      "qkv",
+      "proj",
+      "fc1",
+      "fc2"
+    ],
+    "train_norms": true,
+    "epochs": 1.0,
+    "batch_size": 48,
+    "lr": 0.003,
+    "min_lr": 1e-05,
+    "warmup_frac": 0.1,
+    "betas": [
+      0.9,
+      0.95
+    ],
+    "weight_decay": 0.05,
+    "clip_norm": 4.0,
+    "mixup": 0.0,
+    "mixup_prob": 0.9,
+    "freq_mask": 0,
+    "time_mask": 0,
+    "roll_aug": false,
+    "mask_ratio": 0.0,
+    "num_classes": 200,
+    "prior_bias_init": true,
+    "num_workers": 8,
+    "seed": 0,
+    "train_probe_clips": 4096,
+    "train_eval_every": 25,
+    "val_evals": 5,
+    "eval_batch_size": 128
+  },
+  "gpu": "NVIDIA GeForce RTX 4090",
+  "final_val_map": 0.3877271543070674,
+  "lora/map": 0.3877271543070674,
+  "fsd50k/map": 0.3877271543070674,
+  "train_seconds": 25.9,
+  "total_seconds": 34.92,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.022922640609322115,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.023087197427923888,
+      "val_map": 0.3877271543070674,
+      "lr": 0.0,
+      "train_s": 25.901328155887313
+    }
+  ],
+  "per_class_ap": [
+    0.260008841753006,
+    0.10264948010444641,
+    0.5677452087402344,
+    0.37472274899482727,
+    0.47855356335639954,
+    0.6294817924499512,
+    0.848129391670227,
+    0.569826602935791,
+    0.604072093963623,
+    0.5079339146614075,
+    0.200513556599617,
+    0.44523894786834717,
+    0.13573779165744781,
+    0.3597381114959717,
+    0.5063068270683289,
+    0.4617409408092499,
+    0.036725737154483795,
+    0.246132031083107,
+    0.3643985092639923,
+    0.2549247741699219,
+    0.5793647170066833,
+    0.40309107303619385,
+    0.8263339996337891,
+    0.07593724131584167,
+    0.26588162779808044,
+    0.40415647625923157,
+    0.3906552791595459,
+    0.25304391980171204,
+    0.7883119583129883,
+    0.29423987865448,
+    0.6652907729148865,
+    0.5660180449485779,
+    0.4553625285625458,
+    0.21758468449115753,
+    0.38934534788131714,
+    0.1832878440618515,
+    0.41428855061531067,
+    0.19837094843387604,
+    0.42635586857795715,
+    0.24307867884635925,
+    0.5231850743293762,
+    0.2396785318851471,
+    0.6157076954841614,
+    0.2086513638496399,
+    0.489452064037323,
+    0.1454327255487442,
+    0.26242437958717346,
+    0.251340389251709,
+    0.7089487314224243,
+    0.44370871782302856,
+    0.42789506912231445,
+    0.5127562284469604,
+    0.5684096217155457,
+    0.15242569148540497,
+    0.11684604734182358,
+    0.02989089861512184,
+    0.2682499289512634,
+    0.7270780205726624,
+    0.15121909976005554,
+    0.5215708613395691,
+    0.5919294357299805,
+    0.630686342716217,
+    0.4228959083557129,
+    0.12781782448291779,
+    0.4252147376537323,
+    0.14724557101726532,
+    0.26444771885871887,
+    0.6917716860771179,
+    0.48522427678108215,
+    0.556725800037384,
+    0.4811234474182129,
+    0.21879151463508606,
+    0.6966556906700134,
+    0.3492400646209717,
+    0.3476092517375946,
+    0.24738959968090057,
+    0.4132648706436157,
+    0.4569057524204254,
+    0.24228252470493317,
+    0.44854554533958435,
+    0.2611447870731354,
+    0.3002791106700897,
+    0.2470785230398178,
+    0.3232501447200775,
+    0.197198286652565,
+    0.29001522064208984,
+    0.4156147241592407,
+    0.6822296380996704,
+    0.17385223507881165,
+    0.14074638485908508,
+    0.7612788677215576,
+    0.16763493418693542,
+    0.5093415379524231,
+    0.2034265100955963,
+    0.200764462351799,
+    0.3512108623981476,
+    0.4070063531398773,
+    0.13825556635856628,
+    0.4329879879951477,
+    0.08071372658014297,
+    0.5670169591903687,
+    0.7136991620063782,
+    0.1873439997434616,
+    0.42346805334091187,
+    0.23150715231895447,
+    0.41416409611701965,
+    0.6324176788330078,
+    0.512712836265564,
+    0.561626672744751,
+    0.20138905942440033,
+    0.25556740164756775,
+    0.35534393787384033,
+    0.551257848739624,
+    0.07522853463888168,
+    0.19410252571105957,
+    0.3077264726161957,
+    0.7987044453620911,
+    0.3791520595550537,
+    0.4921835958957672,
+    0.2224937379360199,
+    0.8784657716751099,
+    0.8864381313323975,
+    0.337193101644516,
+    0.16509778797626495,
+    0.2177346646785736,
+    0.7829218506813049,
+    0.28304457664489746,
+    0.7496931552886963,
+    0.33388617634773254,
+    0.14825443923473358,
+    0.1750006377696991,
+    0.7524497509002686,
+    0.4443705081939697,
+    0.42734402418136597,
+    0.63266521692276,
+    0.06515523046255112,
+    0.06078842654824257,
+    0.056424837559461594,
+    0.5956529974937439,
+    0.5499193668365479,
+    0.19116562604904175,
+    0.3281412720680237,
+    0.23890145123004913,
+    0.3512316346168518,
+    0.6900527477264404,
+    0.5790720582008362,
+    0.06491991877555847,
+    0.5748682022094727,
+    0.20201973617076874,
+    0.2559862434864044,
+    0.5126897096633911,
+    0.5108996033668518,
+    0.31607797741889954,
+    0.42942142486572266,
+    0.44624054431915283,
+    0.3004763424396515,
+    0.505363941192627,
+    0.5663438439369202,
+    0.48498523235321045,
+    0.07768086344003677,
+    0.4226090610027313,
+    0.16858841478824615,
+    0.6951196193695068,
+    0.25576454401016235,
+    0.30959251523017883,
+    0.22826725244522095,
+    0.3751213848590851,
+    0.06385413557291031,
+    0.3092803955078125,
+    0.4219374656677246,
+    0.10592648386955261,
+    0.884611964225769,
+    0.8872830867767334,
+    0.05258521810173988,
+    0.5019583702087402,
+    0.7683556079864502,
+    0.20452934503555298,
+    0.2778775691986084,
+    0.1971680223941803,
+    0.3161209523677826,
+    0.06768424063920975,
+    0.6219035983085632,
+    0.13769948482513428,
+    0.5714503526687622,
+    0.6145119071006775,
+    0.4090941250324249,
+    0.2647537291049957,
+    0.6673169136047363,
+    0.5835204124450684,
+    0.36227941513061523,
+    0.38399434089660645,
+    0.4290752410888672,
+    0.5698513388633728,
+    0.11956331878900528,
+    0.2737369239330292,
+    0.38434988260269165,
+    0.10606447607278824,
+    0.471355140209198,
+    0.15620800852775574,
+    0.7690724730491638
+  ]
+}

p4rel-mae-15m-d70-s0/exports/step_00040000/fsd50k_lora_1ep_s1.json

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-15m-d70-s0/exports/step_00040000",
+  "fingerprint": "06edcd5f801cd692",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d70-s0",
+  "target": "fsd50k",
+  "num_classes": 200,
+  "lora_layers_wrapped": 32,
+  "params": {
+    "trainable": 876488,
+    "total": 15158984,
+    "fraction": 0.05781970612278501
+  },
+  "steps_per_epoch": 989,
+  "total_steps": 989,
+  "train_clips": 36796,
+  "train_segments": 47487,
+  "val_clips": 10231,
+  "val_segments": 15513,
+  "train_split": "train",
+  "eval_split": "test",
+  "config": {
+    "rank": 16,
+    "alpha": 32.0,
+    "targets": [
+      "qkv",
+      "proj",
+      "fc1",
+      "fc2"
+    ],
+    "train_norms": true,
+    "epochs": 1.0,
+    "batch_size": 48,
+    "lr": 0.003,
+    "min_lr": 1e-05,
+    "warmup_frac": 0.1,
+    "betas": [
+      0.9,
+      0.95
+    ],
+    "weight_decay": 0.05,
+    "clip_norm": 4.0,
+    "mixup": 0.0,
+    "mixup_prob": 0.9,
+    "freq_mask": 0,
+    "time_mask": 0,
+    "roll_aug": false,
+    "mask_ratio": 0.0,
+    "num_classes": 200,
+    "prior_bias_init": true,
+    "num_workers": 8,
+    "seed": 1,
+    "train_probe_clips": 4096,
+    "train_eval_every": 25,
+    "val_evals": 5,
+    "eval_batch_size": 128
+  },
+  "gpu": "NVIDIA GeForce RTX 4090",
+  "final_val_map": 0.39046065852046014,
+  "lora/map": 0.39046065852046014,
+  "fsd50k/map": 0.39046065852046014,
+  "train_seconds": 27.27,
+  "total_seconds": 36.82,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.0227136660751421,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.022923657324536637,
+      "val_map": 0.39046065852046014,
+      "lr": 0.0,
+      "train_s": 27.27091417997144
+    }
+  ],
+  "per_class_ap": [
+    0.27554604411125183,
+    0.06743300706148148,
+    0.5653056502342224,
+    0.35420772433280945,
+    0.4806189239025116,
+    0.6376400589942932,
+    0.8404497504234314,
+    0.5705803036689758,
+    0.6545842885971069,
+    0.4429802894592285,
+    0.29983073472976685,
+    0.4165874719619751,
+    0.1852922886610031,
+    0.3859442174434662,
+    0.5096136927604675,
+    0.4653187096118927,
+    0.028720341622829437,
+    0.2548602521419525,
+    0.4700181186199188,
+    0.19671127200126648,
+    0.5225346684455872,
+    0.41625887155532837,
+    0.8538342118263245,
+    0.08098524063825607,
+    0.25819483399391174,
+    0.41199740767478943,
+    0.38320407271385193,
+    0.284710556268692,
+    0.7594214081764221,
+    0.30788564682006836,
+    0.6432857513427734,
+    0.6158961057662964,
+    0.4211408793926239,
+    0.17851126194000244,
+    0.37869322299957275,
+    0.23149733245372772,
+    0.42464521527290344,
+    0.1805284172296524,
+    0.28616416454315186,
+    0.3135189414024353,
+    0.5635668039321899,
+    0.23403897881507874,
+    0.5864887833595276,
+    0.22119340300559998,
+    0.5781767964363098,
+    0.19044719636440277,
+    0.27271825075149536,
+    0.2666628360748291,
+    0.6885257363319397,
+    0.44248780608177185,
+    0.40168634057044983,
+    0.48292076587677,
+    0.5255481004714966,
+    0.15990251302719116,
+    0.10975896567106247,
+    0.055185187608003616,
+    0.2910819351673126,
+    0.6933481097221375,
+    0.17377522587776184,
+    0.5354653000831604,
+    0.5883322954177856,
+    0.6396918296813965,
+    0.4345429241657257,
+    0.1463479995727539,
+    0.4898170828819275,
+    0.11341157555580139,
+    0.26276493072509766,
+    0.6689541935920715,
+    0.5359945893287659,
+    0.5390211939811707,
+    0.4943034052848816,
+    0.22818705439567566,
+    0.6907769441604614,
+    0.29292985796928406,
+    0.24998115003108978,
+    0.2376241385936737,
+    0.407619446516037,
+    0.45494046807289124,
+    0.30284976959228516,
+    0.47258448600769043,
+    0.29174092411994934,
+    0.29569336771965027,
+    0.2736303210258484,
+    0.29495179653167725,
+    0.17971886694431305,
+    0.29546141624450684,
+    0.4366110563278198,
+    0.5821285843849182,
+    0.19611963629722595,
+    0.15387104451656342,
+    0.7763524651527405,
+    0.13009777665138245,
+    0.48058411478996277,
+    0.21387313306331635,
+    0.25982749462127686,
+    0.39751869440078735,
+    0.463946670293808,
+    0.20970141887664795,
+    0.3963107168674469,
+    0.06545982509851456,
+    0.5616093277931213,
+    0.7036100625991821,
+    0.22720932960510254,
+    0.44084805250167847,
+    0.2244754135608673,
+    0.4472162127494812,
+    0.6676629781723022,
+    0.5075265169143677,
+    0.5558415055274963,
+    0.23142856359481812,
+    0.26067689061164856,
+    0.35784587264060974,
+    0.5012962222099304,
+    0.11678043007850647,
+    0.1934979110956192,
+    0.35150548815727234,
+    0.671654999256134,
+    0.37167131900787354,
+    0.4950956702232361,
+    0.2685677707195282,
+    0.881673276424408,
+    0.8899623155593872,
+    0.35591405630111694,
+    0.07994064688682556,
+    0.3185991048812866,
+    0.7880043983459473,
+    0.26748666167259216,
+    0.7624571919441223,
+    0.32016074657440186,
+    0.11225560307502747,
+    0.11878355592489243,
+    0.778289794921875,
+    0.5138676166534424,
+    0.47088706493377686,
+    0.6031106114387512,
+    0.08134401589632034,
+    0.07997246086597443,
+    0.06112999841570854,
+    0.6183738708496094,
+    0.562932550907135,
+    0.24265766143798828,
+    0.360340416431427,
+    0.16448336839675903,
+    0.34304195642471313,
+    0.6974566578865051,
+    0.5727390050888062,
+    0.06297561526298523,
+    0.5833868384361267,
+    0.17903479933738708,
+    0.2621118426322937,
+    0.4657540023326874,
+    0.48267269134521484,
+    0.3198612630367279,
+    0.2867625653743744,
+    0.45344460010528564,
+    0.3803417384624481,
+    0.4729243814945221,
+    0.6450055241584778,
+    0.46832573413848877,
+    0.10795183479785919,
+    0.4331681430339813,
+    0.16469641029834747,
+    0.6571678519248962,
+    0.22289682924747467,
+    0.31677544116973877,
+    0.3099585771560669,
+    0.3991954028606415,
+    0.06401166319847107,
+    0.3448405861854553,
+    0.45050591230392456,
+    0.11798356473445892,
+    0.8527421951293945,
+    0.863176167011261,
+    0.04682936519384384,
+    0.5212827324867249,
+    0.7584735751152039,
+    0.18852321803569794,
+    0.27334675192832947,
+    0.21700555086135864,
+    0.2952446937561035,
+    0.04243731498718262,
+    0.4667215347290039,
+    0.20255963504314423,
+    0.6191628575325012,
+    0.6186838746070862,
+    0.4618360698223114,
+    0.25679048895835876,
+    0.6535815596580505,
+    0.618278443813324,
+    0.39008134603500366,
+    0.3607816994190216,
+    0.40643826127052307,
+    0.5652787089347839,
+    0.1313045620918274,
+    0.35181406140327454,
+    0.38263896107673645,
+    0.13142283260822296,
+    0.4529886245727539,
+    0.17486214637756348,
+    0.7618004679679871
+  ]
+}

p4rel-mae-15m-d70-s0/exports/step_00040000/fsd50k_lora_1ep_s2.json

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-15m-d70-s0/exports/step_00040000",
+  "fingerprint": "06edcd5f801cd692",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d70-s0",
+  "target": "fsd50k",
+  "num_classes": 200,
+  "lora_layers_wrapped": 32,
+  "params": {
+    "trainable": 876488,
+    "total": 15158984,
+    "fraction": 0.05781970612278501
+  },
+  "steps_per_epoch": 989,
+  "total_steps": 989,
+  "train_clips": 36796,
+  "train_segments": 47487,
+  "val_clips": 10231,
+  "val_segments": 15513,
+  "train_split": "train",
+  "eval_split": "test",
+  "config": {
+    "rank": 16,
+    "alpha": 32.0,
+    "targets": [
+      "qkv",
+      "proj",
+      "fc1",
+      "fc2"
+    ],
+    "train_norms": true,
+    "epochs": 1.0,
+    "batch_size": 48,
+    "lr": 0.003,
+    "min_lr": 1e-05,
+    "warmup_frac": 0.1,
+    "betas": [
+      0.9,
+      0.95
+    ],
+    "weight_decay": 0.05,
+    "clip_norm": 4.0,
+    "mixup": 0.0,
+    "mixup_prob": 0.9,
+    "freq_mask": 0,
+    "time_mask": 0,
+    "roll_aug": false,
+    "mask_ratio": 0.0,
+    "num_classes": 200,
+    "prior_bias_init": true,
+    "num_workers": 8,
+    "seed": 2,
+    "train_probe_clips": 4096,
+    "train_eval_every": 25,
+    "val_evals": 5,
+    "eval_batch_size": 128
+  },
+  "gpu": "NVIDIA GeForce RTX 4090",
+  "final_val_map": 0.3868309912178665,
+  "lora/map": 0.3868309912178665,
+  "fsd50k/map": 0.3868309912178665,
+  "train_seconds": 27.29,
+  "total_seconds": 36.41,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.0224177177506499,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.023292145480979822,
+      "val_map": 0.3868309912178665,
+      "lr": 0.0,
+      "train_s": 27.285456705954857
+    }
+  ],
+  "per_class_ap": [
+    0.22061371803283691,
+    0.1078604906797409,
+    0.5660447478294373,
+    0.376985102891922,
+    0.48863837122917175,
+    0.6314563155174255,
+    0.8528037071228027,
+    0.5623655319213867,
+    0.6280444264411926,
+    0.45181724429130554,
+    0.2678086757659912,
+    0.43013420701026917,
+    0.12419147789478302,
+    0.44320887327194214,
+    0.48801925778388977,
+    0.46082934737205505,
+    0.04619644582271576,
+    0.20013117790222168,
+    0.45107486844062805,
+    0.23290544748306274,
+    0.5144052505493164,
+    0.41405585408210754,
+    0.8636059165000916,
+    0.11878174543380737,
+    0.2662009298801422,
+    0.4578126072883606,
+    0.3521581292152405,
+    0.2756655514240265,
+    0.7721804976463318,
+    0.2928084433078766,
+    0.6725174188613892,
+    0.5907047390937805,
+    0.40503889322280884,
+    0.22111955285072327,
+    0.4211660623550415,
+    0.20004285871982574,
+    0.40480852127075195,
+    0.16597993671894073,
+    0.3120489716529846,
+    0.29344287514686584,
+    0.5269410610198975,
+    0.1919204443693161,
+    0.5473778247833252,
+    0.21163776516914368,
+    0.5214588642120361,
+    0.15349291265010834,
+    0.25205811858177185,
+    0.28005465865135193,
+    0.7379135489463806,
+    0.441096693277359,
+    0.3571303188800812,
+    0.500606894493103,
+    0.5392037034034729,
+    0.1337556391954422,
+    0.18934814631938934,
+    0.015780074521899223,
+    0.21463464200496674,
+    0.6838313937187195,
+    0.13308477401733398,
+    0.5064087510108948,
+    0.5695154666900635,
+    0.625920832157135,
+    0.4242378771305084,
+    0.2345944494009018,
+    0.43885695934295654,
+    0.1197628527879715,
+    0.22726145386695862,
+    0.6441606879234314,
+    0.48017728328704834,
+    0.5708172917366028,
+    0.48092007637023926,
+    0.22765842080116272,
+    0.7033281922340393,
+    0.32898107171058655,
+    0.28890132904052734,
+    0.25301170349121094,
+    0.451901376247406,
+    0.48233145475387573,
+    0.28056445717811584,
+    0.5200304388999939,
+    0.22746583819389343,
+    0.2375781387090683,
+    0.28260642290115356,
+    0.2154625654220581,
+    0.23780444264411926,
+    0.27289441227912903,
+    0.4221517741680145,
+    0.6816415786743164,
+    0.3161788582801819,
+    0.08656533807516098,
+    0.7551863789558411,
+    0.10895133763551712,
+    0.5416502356529236,
+    0.17363402247428894,
+    0.3089142441749573,
+    0.37526991963386536,
+    0.33939921855926514,
+    0.1778591275215149,
+    0.4089067578315735,
+    0.07862019538879395,
+    0.5572622418403625,
+    0.7176440954208374,
+    0.19862833619117737,
+    0.4296669661998749,
+    0.25769931077957153,
+    0.3744755685329437,
+    0.6532266736030579,
+    0.5005418658256531,
+    0.5525277853012085,
+    0.20239001512527466,
+    0.26122069358825684,
+    0.3394911289215088,
+    0.5685809254646301,
+    0.07271594554185867,
+    0.17205046117305756,
+    0.3187377452850342,
+    0.7483323216438293,
+    0.42232298851013184,
+    0.4961910843849182,
+    0.23510506749153137,
+    0.8723099231719971,
+    0.8814102411270142,
+    0.36920690536499023,
+    0.1455078423023224,
+    0.19192728400230408,
+    0.7702342867851257,
+    0.3740755319595337,
+    0.7434014081954956,
+    0.34938931465148926,
+    0.10895656049251556,
+    0.19754205644130707,
+    0.7466550469398499,
+    0.5084183812141418,
+    0.4731442332267761,
+    0.5993091464042664,
+    0.053749535232782364,
+    0.09687972813844681,
+    0.0713169053196907,
+    0.6692836880683899,
+    0.5638151168823242,
+    0.16735400259494781,
+    0.3009190261363983,
+    0.2163551300764084,
+    0.37173959612846375,
+    0.6669848561286926,
+    0.5665475130081177,
+    0.06883008033037186,
+    0.5985261797904968,
+    0.19075770676136017,
+    0.2552443742752075,
+    0.5275648236274719,
+    0.460819810628891,
+    0.3054390549659729,
+    0.3483526408672333,
+    0.4320860803127289,
+    0.2616671919822693,
+    0.5473411679267883,
+    0.6324376463890076,
+    0.4688871204853058,
+    0.05515439435839653,
+    0.43563732504844666,
+    0.16903461515903473,
+    0.6495463252067566,
+    0.2829745411872864,
+    0.34481680393218994,
+    0.33904048800468445,
+    0.3403007984161377,
+    0.06233499199151993,
+    0.3036753237247467,
+    0.41298308968544006,
+    0.11084333807229996,
+    0.8384218811988831,
+    0.8439898490905762,
+    0.045269884169101715,
+    0.5186144709587097,
+    0.7559388279914856,
+    0.1736414134502411,
+    0.2861306965351105,
+    0.20177704095840454,
+    0.31930726766586304,
+    0.04537332430481911,
+    0.45097896456718445,
+    0.07063556462526321,
+    0.5433626770973206,
+    0.6072033047676086,
+    0.4015720784664154,
+    0.275083988904953,
+    0.6626448631286621,
+    0.5736453533172607,
+    0.41733071208000183,
+    0.4724583923816681,
+    0.4504481852054596,
+    0.5594882369041443,
+    0.13681946694850922,
+    0.2988184690475464,
+    0.39722558856010437,
+    0.11067724227905273,
+    0.4273834228515625,
+    0.16706006228923798,
+    0.7702717185020447
+  ]
+}

p4rel-mae-15m-d70-s0/exports/step_00040000/lora_1ep_s0.json

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-15m-d70-s0/exports/step_00040000",
+    "fingerprint": "06edcd5f801cd692",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-15m-d70-s0",
+    "lora_layers_wrapped": 32,
+    "params": {
+      "trainable": 1002383,
+      "total": 15284879,
+      "fraction": 0.0655800415561026
+    },
+    "steps_per_epoch": 428,
+    "total_steps": 428,
+    "train_clips": 20550,
+    "val_clips": 18886,
+    "config": {
+      "rank": 16,
+      "alpha": 32.0,
+      "targets": [
+        "qkv",
+        "proj",
+        "fc1",
+        "fc2"
+      ],
+      "train_norms": true,
+      "epochs": 1.0,
+      "batch_size": 48,
+      "lr": 0.003,
+      "min_lr": 1e-05,
+      "warmup_frac": 0.1,
+      "betas": [
+        0.9,
+        0.95
+      ],
+      "weight_decay": 0.05,
+      "clip_norm": 4.0,
+      "mixup": 0.0,
+      "mixup_prob": 0.9,
+      "freq_mask": 0,
+      "time_mask": 0,
+      "roll_aug": false,
+      "mask_ratio": 0.0,
+      "num_classes": 527,
+      "prior_bias_init": true,
+      "num_workers": 8,
+      "seed": 0,
+      "train_probe_clips": 4096,
+      "train_eval_every": 25,
+      "val_evals": 5,
+      "eval_batch_size": 128
+    },
+    "gpu": "NVIDIA GeForce RTX 4090"
+  },
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "train_map": 0.0074939395933265335,
+      "val_map": 0.005712181667928041,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02358,
+      "train_map": 0.03080750084550613,
+      "val_map": null,
+      "lr": 0.0017857142857142857,
+      "train_s": 0.9
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.02297,
+      "train_map": 0.05707648851417457,
+      "val_map": null,
+      "lr": 0.002996832162669313,
+      "train_s": 1.5
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02232,
+      "train_map": 0.06999237047936026,
+      "val_map": null,
+      "lr": 0.0029464015850084926,
+      "train_s": 2.2
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02184,
+      "train_map": 0.08950785380846465,
+      "val_map": null,
+      "lr": 0.0028365019165406584,
+      "train_s": 2.8
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02187,
+      "train_map": 0.08909362465968493,
+      "val_map": 0.060702448080345084,
+      "lr": 0.002795633084492174,
+      "train_s": 3.0
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02071,
+      "train_map": 0.10208499975824124,
+      "val_map": null,
+      "lr": 0.0026716673782849554,
+      "train_s": 3.5
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02018,
+      "train_map": 0.11399584567833487,
+      "val_map": null,
+      "lr": 0.002458698684369641,
+      "train_s": 4.2
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.02007,
+      "train_map": 0.13578757894114085,
+      "val_map": null,
+      "lr": 0.0022063824593754194,
+      "train_s": 4.9
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.01977,
+      "train_map": 0.15018134654200932,
+      "val_map": null,
+      "lr": 0.0019251287213377403,
+      "train_s": 5.6
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.01911,
+      "train_map": 0.1576283530798693,
+      "val_map": 0.09847784349677176,
+      "lr": 0.0017592766878384548,
+      "train_s": 6.0
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01947,
+      "train_map": 0.16120263626816658,
+      "val_map": null,
+      "lr": 0.0016265413870700269,
+      "train_s": 6.3
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.01923,
+      "train_map": 0.18461839072105865,
+      "val_map": null,
+      "lr": 0.0013229395198229876,
+      "train_s": 6.9
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01902,
+      "train_map": 0.191946903617427,
+      "val_map": null,
+      "lr": 0.0010268490715591258,
+      "train_s": 7.6
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01857,
+      "train_map": 0.20945366083925318,
+      "val_map": null,
+      "lr": 0.000750486089449494,
+      "train_s": 8.3
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.01798,
+      "train_map": 0.21804560740625015,
+      "val_map": 0.13428084322461678,
+      "lr": 0.000541962785848462,
+      "train_s": 8.9
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01834,
+      "train_map": 0.21708996694683463,
+      "val_map": null,
+      "lr": 0.0005052527083673369,
+      "train_s": 9.0
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01775,
+      "train_map": 0.22611872359325638,
+      "val_map": null,
+      "lr": 0.00030126672363088163,
+      "train_s": 9.6
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01725,
+      "train_map": 0.23505289831589307,
+      "val_map": null,
+      "lr": 0.00014694415279957929,
+      "train_s": 10.3
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01695,
+      "train_map": 0.23854757251263908,
+      "val_map": null,
+      "lr": 4.865200911511649e-05,
+      "train_s": 10.9
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.01754,
+      "train_map": 0.2391753563378183,
+      "val_map": null,
+      "lr": 1.0445612398966596e-05,
+      "train_s": 11.5
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.01763,
+      "train_map": 0.23939312182091163,
+      "val_map": 0.14809199541354767,
+      "lr": 1e-05,
+      "train_s": 11.6
+    }
+  ],
+  "final_train_map": 0.23939312182091163,
+  "final_val_map": 0.14809199541354767,
+  "lora/map": 0.14809199541354767,
+  "train_seconds": 11.6,
+  "wall_seconds": 64.5
+}

p4rel-mae-15m-d70-s0/exports/step_00040000/lora_1ep_s1.json

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-15m-d70-s0/exports/step_00040000",
+    "fingerprint": "06edcd5f801cd692",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-15m-d70-s0",
+    "lora_layers_wrapped": 32,
+    "params": {
+      "trainable": 1002383,
+      "total": 15284879,
+      "fraction": 0.0655800415561026
+    },
+    "steps_per_epoch": 428,
+    "total_steps": 428,
+    "train_clips": 20550,
+    "val_clips": 18886,
+    "config": {
+      "rank": 16,
+      "alpha": 32.0,
+      "targets": [
+        "qkv",
+        "proj",
+        "fc1",
+        "fc2"
+      ],
+      "train_norms": true,
+      "epochs": 1.0,
+      "batch_size": 48,
+      "lr": 0.003,
+      "min_lr": 1e-05,
+      "warmup_frac": 0.1,
+      "betas": [
+        0.9,
+        0.95
+      ],
+      "weight_decay": 0.05,
+      "clip_norm": 4.0,
+      "mixup": 0.0,
+      "mixup_prob": 0.9,
+      "freq_mask": 0,
+      "time_mask": 0,
+      "roll_aug": false,
+      "mask_ratio": 0.0,
+      "num_classes": 527,
+      "prior_bias_init": true,
+      "num_workers": 8,
+      "seed": 1,
+      "train_probe_clips": 4096,
+      "train_eval_every": 25,
+      "val_evals": 5,
+      "eval_batch_size": 128
+    },
+    "gpu": "NVIDIA GeForce RTX 4090"
+  },
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "train_map": 0.007049383063874856,
+      "val_map": 0.005938519821131153,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02449,
+      "train_map": 0.039466586422335984,
+      "val_map": null,
+      "lr": 0.0017857142857142857,
+      "train_s": 0.7
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.02282,
+      "train_map": 0.059709048583556634,
+      "val_map": null,
+      "lr": 0.002996832162669313,
+      "train_s": 1.4
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02199,
+      "train_map": 0.07217455250290584,
+      "val_map": null,
+      "lr": 0.0029464015850084926,
+      "train_s": 2.1
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02133,
+      "train_map": 0.09204399516890832,
+      "val_map": null,
+      "lr": 0.0028365019165406584,
+      "train_s": 2.7
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02157,
+      "train_map": 0.09233808517456055,
+      "val_map": 0.06294601598570734,
+      "lr": 0.002795633084492174,
+      "train_s": 2.9
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02072,
+      "train_map": 0.1052533306863576,
+      "val_map": null,
+      "lr": 0.0026716673782849554,
+      "train_s": 3.4
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02062,
+      "train_map": 0.12061589544699076,
+      "val_map": null,
+      "lr": 0.002458698684369641,
+      "train_s": 4.1
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.01998,
+      "train_map": 0.12750337119576516,
+      "val_map": null,
+      "lr": 0.0022063824593754194,
+      "train_s": 4.7
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.02028,
+      "train_map": 0.14515177830187798,
+      "val_map": null,
+      "lr": 0.0019251287213377403,
+      "train_s": 5.4
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.01945,
+      "train_map": 0.1587755172124638,
+      "val_map": 0.09370038350001401,
+      "lr": 0.0017592766878384548,
+      "train_s": 5.8
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.0195,
+      "train_map": 0.16312486467091278,
+      "val_map": null,
+      "lr": 0.0016265413870700269,
+      "train_s": 6.1
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.01905,
+      "train_map": 0.17228526957297743,
+      "val_map": null,
+      "lr": 0.0013229395198229876,
+      "train_s": 6.7
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01899,
+      "train_map": 0.1943365976027084,
+      "val_map": null,
+      "lr": 0.0010268490715591258,
+      "train_s": 7.4
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01832,
+      "train_map": 0.20845775567507635,
+      "val_map": null,
+      "lr": 0.000750486089449494,
+      "train_s": 8.1
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.01787,
+      "train_map": 0.22036512249080462,
+      "val_map": 0.13493579515455684,
+      "lr": 0.000541962785848462,
+      "train_s": 8.6
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01818,
+      "train_map": 0.22200987797249921,
+      "val_map": null,
+      "lr": 0.0005052527083673369,
+      "train_s": 8.7
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01748,
+      "train_map": 0.23145257725170063,
+      "val_map": null,
+      "lr": 0.00030126672363088163,
+      "train_s": 9.4
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01785,
+      "train_map": 0.2366878639604451,
+      "val_map": null,
+      "lr": 0.00014694415279957929,
+      "train_s": 10.1
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01742,
+      "train_map": 0.23880295984206662,
+      "val_map": null,
+      "lr": 4.865200911511649e-05,
+      "train_s": 10.7
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.01746,
+      "train_map": 0.24051235490027936,
+      "val_map": null,
+      "lr": 1.0445612398966596e-05,
+      "train_s": 11.3
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.01725,
+      "train_map": 0.24037582423552106,
+      "val_map": 0.14980901386217696,
+      "lr": 1e-05,
+      "train_s": 11.4
+    }
+  ],
+  "final_train_map": 0.24037582423552106,
+  "final_val_map": 0.14980901386217696,
+  "lora/map": 0.14980901386217696,
+  "train_seconds": 11.4,
+  "wall_seconds": 62.4
+}

p4rel-mae-15m-d70-s0/exports/step_00040000/lora_1ep_s2.json

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-15m-d70-s0/exports/step_00040000",
+    "fingerprint": "06edcd5f801cd692",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-15m-d70-s0",
+    "lora_layers_wrapped": 32,
+    "params": {
+      "trainable": 1002383,
+      "total": 15284879,
+      "fraction": 0.0655800415561026
+    },
+    "steps_per_epoch": 428,
+    "total_steps": 428,
+    "train_clips": 20550,
+    "val_clips": 18886,
+    "config": {
+      "rank": 16,
+      "alpha": 32.0,
+      "targets": [
+        "qkv",
+        "proj",
+        "fc1",
+        "fc2"
+      ],
+      "train_norms": true,
+      "epochs": 1.0,
+      "batch_size": 48,
+      "lr": 0.003,
+      "min_lr": 1e-05,
+      "warmup_frac": 0.1,
+      "betas": [
+        0.9,
+        0.95
+      ],
+      "weight_decay": 0.05,
+      "clip_norm": 4.0,
+      "mixup": 0.0,
+      "mixup_prob": 0.9,
+      "freq_mask": 0,
+      "time_mask": 0,
+      "roll_aug": false,
+      "mask_ratio": 0.0,
+      "num_classes": 527,
+      "prior_bias_init": true,
+      "num_workers": 8,
+      "seed": 2,
+      "train_probe_clips": 4096,
+      "train_eval_every": 25,
+      "val_evals": 5,
+      "eval_batch_size": 128
+    },
+    "gpu": "NVIDIA GeForce RTX 4090"
+  },
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "train_map": 0.007046422147191178,
+      "val_map": 0.006106081695924999,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02402,
+      "train_map": 0.031653309878239326,
+      "val_map": null,
+      "lr": 0.0017857142857142857,
+      "train_s": 0.7
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.02335,
+      "train_map": 0.04185013510388423,
+      "val_map": null,
+      "lr": 0.002996832162669313,
+      "train_s": 1.4
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02217,
+      "train_map": 0.05635314939085896,
+      "val_map": null,
+      "lr": 0.0029464015850084926,
+      "train_s": 2.1
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.0212,
+      "train_map": 0.08065614166764093,
+      "val_map": null,
+      "lr": 0.0028365019165406584,
+      "train_s": 2.8
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02165,
+      "train_map": 0.08267354467827155,
+      "val_map": 0.05500135935871266,
+      "lr": 0.002795633084492174,
+      "train_s": 3.0
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02134,
+      "train_map": 0.09090852386369397,
+      "val_map": null,
+      "lr": 0.0026716673782849554,
+      "train_s": 3.4
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02105,
+      "train_map": 0.10123847607272259,
+      "val_map": null,
+      "lr": 0.002458698684369641,
+      "train_s": 4.1
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.02055,
+      "train_map": 0.12925484099653345,
+      "val_map": null,
+      "lr": 0.0022063824593754194,
+      "train_s": 4.8
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.01936,
+      "train_map": 0.1414052431340872,
+      "val_map": null,
+      "lr": 0.0019251287213377403,
+      "train_s": 5.4
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.01891,
+      "train_map": 0.15255539260532502,
+      "val_map": 0.09592937589384459,
+      "lr": 0.0017592766878384548,
+      "train_s": 5.8
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01898,
+      "train_map": 0.15605298093426442,
+      "val_map": null,
+      "lr": 0.0016265413870700269,
+      "train_s": 6.1
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.01876,
+      "train_map": 0.16972427966649267,
+      "val_map": null,
+      "lr": 0.0013229395198229876,
+      "train_s": 6.8
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01888,
+      "train_map": 0.18178030891093006,
+      "val_map": null,
+      "lr": 0.0010268490715591258,
+      "train_s": 7.5
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01835,
+      "train_map": 0.19247777140045458,
+      "val_map": null,
+      "lr": 0.000750486089449494,
+      "train_s": 8.2
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.01822,
+      "train_map": 0.20784303802728257,
+      "val_map": 0.13044211801417185,
+      "lr": 0.000541962785848462,
+      "train_s": 8.7
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01793,
+      "train_map": 0.2092737408745433,
+      "val_map": null,
+      "lr": 0.0005052527083673369,
+      "train_s": 8.8
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01793,
+      "train_map": 0.21976662398292993,
+      "val_map": null,
+      "lr": 0.00030126672363088163,
+      "train_s": 9.5
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01789,
+      "train_map": 0.22545952269934696,
+      "val_map": null,
+      "lr": 0.00014694415279957929,
+      "train_s": 10.1
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01769,
+      "train_map": 0.2304362047205642,
+      "val_map": null,
+      "lr": 4.865200911511649e-05,
+      "train_s": 10.8
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.01765,
+      "train_map": 0.23151717669519628,
+      "val_map": null,
+      "lr": 1.0445612398966596e-05,
+      "train_s": 11.4
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.01802,
+      "train_map": 0.2313127115269879,
+      "val_map": 0.14415238846616513,
+      "lr": 1e-05,
+      "train_s": 11.5
+    }
+  ],
+  "final_train_map": 0.2313127115269879,
+  "final_val_map": 0.14415238846616513,
+  "lora/map": 0.14415238846616513,
+  "train_seconds": 11.5,
+  "wall_seconds": 64.8
+}

p4rel-mae-15m-d70-s0/exports/step_00040000/probe.json

@@ -0,0 +1,576 @@
+{
+  "export_dir": "/workspace/runs/p4rel-mae-15m-d70-s0/exports/step_00040000",
+  "fingerprint": "06edcd5f801cd692",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d70-s0",
+  "probe/map": 0.16324810734092907,
+  "probe/map_train": 0.514787645787849,
+  "probe/alpha": 1.0,
+  "probe/holdout_map": 0.19121946726017322,
+  "probe/pooling": "concat",
+  "probe/feature_dim": 768,
+  "probe/fit_clips": 20550,
+  "probe/score_clips": 18886,
+  "probe/classes_scored": 527,
+  "probe/extract_s": 7.259110993996728,
+  "probe/total_s": 8.782387015002314,
+  "alpha_selection": [
+    {
+      "alpha": 0.01,
+      "holdout_map": 0.1908032312455536
+    },
+    {
+      "alpha": 0.1,
+      "holdout_map": 0.19080726282964453
+    },
+    {
+      "alpha": 1.0,
+      "holdout_map": 0.19121946726017322
+    },
+    {
+      "alpha": 10.0,
+      "holdout_map": 0.18788772823193606
+    },
+    {
+      "alpha": 100.0,
+      "holdout_map": 0.17496996812444654
+    },
+    {
+      "alpha": 1000.0,
+      "holdout_map": 0.1502742491224233
+    },
+    {
+      "alpha": 10000.0,
+      "holdout_map": 0.11776263614501487
+    }
+  ],
+  "per_class_ap": [
+    0.6573300957679749,
+    0.02367023378610611,
+    0.07788864523172379,
+    0.22934624552726746,
+    0.14310158789157867,
+    0.275036096572876,
+    0.24697604775428772,
+    0.3112850785255432,
+    0.0484897755086422,
+    0.04415246844291687,
+    0.024010363966226578,
+    0.09299605339765549,
+    0.5629262328147888,
+    0.3598296642303467,
+    0.05725002661347389,
+    0.5587634444236755,
+    0.32977697253227234,
+    0.3732767701148987,
+    0.07235199958086014,
+    0.12791705131530762,
+    0.27237629890441895,
+    0.08199446648359299,
+    0.24110504984855652,
+    0.15162554383277893,
+    0.10530249774456024,
+    0.0235089510679245,
+    0.12746946513652802,
+    0.19934415817260742,
+    0.3319665193557739,
+    0.1350327581167221,
+    0.2548009157180786,
+    0.2894749045372009,
+    0.036798808723688126,
+    0.08206506073474884,
+    0.27125686407089233,
+    0.07517223805189133,
+    0.4388969838619232,
+    0.07100217044353485,
+    0.07299548387527466,
+    0.18241636455059052,
+    0.16563668847084045,
+    0.028134997934103012,
+    0.08724837750196457,
+    0.513539731502533,
+    0.047474343329668045,
+    0.11499907076358795,
+    0.01944059506058693,
+    0.14457817375659943,
+    0.06532446295022964,
+    0.05966693535447121,
+    0.15606547892093658,
+    0.01471930742263794,
+    0.034413326531648636,
+    0.08448060601949692,
+    0.258444607257843,
+    0.12523241341114044,
+    0.26651492714881897,
+    0.36561787128448486,
+    0.1532941460609436,
+    0.08591976016759872,
+    0.20324018597602844,
+    0.23854577541351318,
+    0.07430237531661987,
+    0.20401164889335632,
+    0.544398844242096,
+    0.23772871494293213,
+    0.3090175688266754,
+    0.3544028401374817,
+    0.4099395275115967,
+    0.41064274311065674,
+    0.1530970185995102,
+    0.15974406898021698,
+    0.2340248078107834,
+    0.21949037909507751,
+    0.2243235558271408,
+    0.2616903483867645,
+    0.060182999819517136,
+    0.24284376204013824,
+    0.2037993222475052,
+    0.14395703375339508,
+    0.019984720274806023,
+    0.18220365047454834,
+    0.40476125478744507,
+    0.2284671813249588,
+    0.114028200507164,
+    0.043899327516555786,
+    0.09535790234804153,
+    0.035932593047618866,
+    0.03200475871562958,
+    0.06792169064283371,
+    0.10381541401147842,
+    0.09755466878414154,
+    0.20629353821277618,
+    0.039485882967710495,
+    0.04175325483083725,
+    0.09264999628067017,
+    0.022577255964279175,
+    0.02591581828892231,
+    0.15460485219955444,
+    0.1894698143005371,
+    0.15793059766292572,
+    0.29079657793045044,
+    0.05482707917690277,
+    0.05779309198260307,
+    0.029136385768651962,
+    0.06332720816135406,
+    0.07536691427230835,
+    0.08318232744932175,
+    0.11172317713499069,
+    0.20181439816951752,
+    0.2668752074241638,
+    0.16927261650562286,
+    0.16064634919166565,
+    0.11072859913110733,
+    0.07319800555706024,
+    0.25229695439338684,
+    0.2491992712020874,
+    0.1613241583108902,
+    0.14892074465751648,
+    0.3660365045070648,
+    0.3717231750488281,
+    0.1590077131986618,
+    0.0875227227807045,
+    0.1133202388882637,
+    0.015510759316384792,
+    0.03287111595273018,
+    0.15553437173366547,
+    0.18352974951267242,
+    0.21394792199134827,
+    0.197184756398201,
+    0.006231526844203472,
+    0.12980403006076813,
+    0.252254843711853,
+    0.2459551990032196,
+    0.06788132339715958,
+    0.011671967804431915,
+    0.11910194158554077,
+    0.7391643524169922,
+    0.2586532533168793,
+    0.3586351275444031,
+    0.4416726529598236,
+    0.3191075623035431,
+    0.20186349749565125,
+    0.19221387803554535,
+    0.11193602532148361,
+    0.20638921856880188,
+    0.16402602195739746,
+    0.3311121463775635,
+    0.20523110032081604,
+    0.26594799757003784,
+    0.15791666507720947,
+    0.17431621253490448,
+    0.289266973733902,
+    0.14488500356674194,
+    0.13472551107406616,
+    0.2730728089809418,
+    0.17480270564556122,
+    0.13480769097805023,
+    0.03977874666452408,
+    0.08078286051750183,
+    0.5119502544403076,
+    0.3122377097606659,
+    0.3202573359012604,
+    0.17231427133083344,
+    0.28146839141845703,
+    0.29405057430267334,
+    0.31291452050209045,
+    0.1419290006160736,
+    0.30175480246543884,
+    0.4484557509422302,
+    0.299505352973938,
+    0.2919064164161682,
+    0.3205241858959198,
+    0.21407128870487213,
+    0.18574368953704834,
+    0.13689455389976501,
+    0.175179585814476,
+    0.36408647894859314,
+    0.07576873153448105,
+    0.21300804615020752,
+    0.2668962776660919,
+    0.25283604860305786,
+    0.14841483533382416,
+    0.3197208046913147,
+    0.08465944230556488,
+    0.4615556001663208,
+    0.39913812279701233,
+    0.20572882890701294,
+    0.34799134731292725,
+    0.20177331566810608,
+    0.19015546143054962,
+    0.13129642605781555,
+    0.3309966027736664,
+    0.2676827907562256,
+    0.252322793006897,
+    0.4009498059749603,
+    0.07056787610054016,
+    0.12097745388746262,
+    0.15010616183280945,
+    0.3088320791721344,
+    0.046635109931230545,
+    0.20736251771450043,
+    0.11116752028465271,
+    0.13623438775539398,
+    0.23038366436958313,
+    0.21212340891361237,
+    0.24441762268543243,
+    0.5804275274276733,
+    0.2509101629257202,
+    0.373256117105484,
+    0.7431393265724182,
+    0.501549482345581,
+    0.5271211862564087,
+    0.059925828129053116,
+    0.3429856598377228,
+    0.18978236615657806,
+    0.05640650540590286,
+    0.3099146783351898,
+    0.2101779729127884,
+    0.16585205495357513,
+    0.19441474974155426,
+    0.14866943657398224,
+    0.17430292069911957,
+    0.23412449657917023,
+    0.12615013122558594,
+    0.11710449308156967,
+    0.07617956399917603,
+    0.03993690386414528,
+    0.15868347883224487,
+    0.053605105727910995,
+    0.07692670077085495,
+    0.07901681214570999,
+    0.0802655816078186,
+    0.05193301662802696,
+    0.0800684243440628,
+    0.021086934953927994,
+    0.19410032033920288,
+    0.02288595587015152,
+    0.40587690472602844,
+    0.03560571372509003,
+    0.14759598672389984,
+    0.22915714979171753,
+    0.2844068706035614,
+    0.16903553903102875,
+    0.1833532601594925,
+    0.2865458130836487,
+    0.08238542079925537,
+    0.15222875773906708,
+    0.032521940767765045,
+    0.11200763285160065,
+    0.1409483551979065,
+    0.019084392115473747,
+    0.1123746931552887,
+    0.08301116526126862,
+    0.07944425195455551,
+    0.2260444462299347,
+    0.25329795479774475,
+    0.25080448389053345,
+    0.16545020043849945,
+    0.11422935873270035,
+    0.04580200836062431,
+    0.12508101761341095,
+    0.2282620221376419,
+    0.08705272525548935,
+    0.022114092484116554,
+    0.07472071796655655,
+    0.08627159893512726,
+    0.17246170341968536,
+    0.1693018227815628,
+    0.03943503275513649,
+    0.02975887432694435,
+    0.2758210599422455,
+    0.19288785755634308,
+    0.07579165697097778,
+    0.23196503520011902,
+    0.042307570576667786,
+    0.06887106597423553,
+    0.05997755378484726,
+    0.07967521995306015,
+    0.06631679087877274,
+    0.027806198224425316,
+    0.29075488448143005,
+    0.13928082585334778,
+    0.24811729788780212,
+    0.08079108595848083,
+    0.2716726064682007,
+    0.3646240830421448,
+    0.3764822483062744,
+    0.2616892158985138,
+    0.25402843952178955,
+    0.23050391674041748,
+    0.2165817767381668,
+    0.1202303022146225,
+    0.21149638295173645,
+    0.2337527722120285,
+    0.24826733767986298,
+    0.1775970607995987,
+    0.009156990796327591,
+    0.011168018914759159,
+    0.0171671099960804,
+    0.30264315009117126,
+    0.2164454311132431,
+    0.2030855119228363,
+    0.08266929537057877,
+    0.04587468132376671,
+    0.11843400448560715,
+    0.02755981869995594,
+    0.19786782562732697,
+    0.07603122293949127,
+    0.17155343294143677,
+    0.15662550926208496,
+    0.09554506093263626,
+    0.1344657838344574,
+    0.0981256291270256,
+    0.07854066044092178,
+    0.1630316972732544,
+    0.08233069628477097,
+    0.07066692411899567,
+    0.16622571647167206,
+    0.09233389049768448,
+    0.34946590662002563,
+    0.051375217735767365,
+    0.4678316116333008,
+    0.45391011238098145,
+    0.2750501036643982,
+    0.26704633235931396,
+    0.08395764231681824,
+    0.08066295087337494,
+    0.2806853950023651,
+    0.26465949416160583,
+    0.1440567672252655,
+    0.24039383232593536,
+    0.28871020674705505,
+    0.3566828966140747,
+    0.20852181315422058,
+    0.21182173490524292,
+    0.19576382637023926,
+    0.24750718474388123,
+    0.06776642054319382,
+    0.050274260342121124,
+    0.1578582376241684,
+    0.007586134597659111,
+    0.14398755133152008,
+    0.15555395185947418,
+    0.08852828294038773,
+    0.07484275847673416,
+    0.13191623985767365,
+    0.33409741520881653,
+    0.036808714270591736,
+    0.02964021824300289,
+    0.13521426916122437,
+    0.037687934935092926,
+    0.10444795340299606,
+    0.11061224341392517,
+    0.043817829340696335,
+    0.05697917193174362,
+    0.03253432735800743,
+    0.013980414718389511,
+    0.037564314901828766,
+    0.13073086738586426,
+    0.016647865995764732,
+    0.07868801057338715,
+    0.15325020253658295,
+    0.16144143044948578,
+    0.035702019929885864,
+    0.05829491466283798,
+    0.41699856519699097,
+    0.20666874945163727,
+    0.019482532516121864,
+    0.10932868719100952,
+    0.15345259010791779,
+    0.09167686849832535,
+    0.1713397204875946,
+    0.3475460410118103,
+    0.18148523569107056,
+    0.0845271572470665,
+    0.13030119240283966,
+    0.0868024155497551,
+    0.021539457142353058,
+    0.01354830153286457,
+    0.02932128868997097,
+    0.0992605909705162,
+    0.19527263939380646,
+    0.08060897886753082,
+    0.20726922154426575,
+    0.2054988592863083,
+    0.17541657388210297,
+    0.07193592190742493,
+    0.05597496032714844,
+    0.05618658289313316,
+    0.2285613864660263,
+    0.14503145217895508,
+    0.055630046874284744,
+    0.3546653091907501,
+    0.2613515257835388,
+    0.10317108780145645,
+    0.5006682276725769,
+    0.5453035831451416,
+    0.027333658188581467,
+    0.3798114061355591,
+    0.443663090467453,
+    0.0580679215490818,
+    0.12386137247085571,
+    0.2679891288280487,
+    0.031883783638477325,
+    0.01613554172217846,
+    0.04347144439816475,
+    0.04149666428565979,
+    0.061458319425582886,
+    0.018784409388899803,
+    0.01863982155919075,
+    0.14376574754714966,
+    0.027220509946346283,
+    0.0654742568731308,
+    0.27982649207115173,
+    0.20677867531776428,
+    0.05791512876749039,
+    0.021104415878653526,
+    0.12983988225460052,
+    0.09419628977775574,
+    0.27732908725738525,
+    0.08172032982110977,
+    0.11654742807149887,
+    0.3168684244155884,
+    0.1299894005060196,
+    0.13132844865322113,
+    0.05781286582350731,
+    0.22771620750427246,
+    0.2977781593799591,
+    0.16098511219024658,
+    0.10157264024019241,
+    0.1799076646566391,
+    0.17544400691986084,
+    0.06881210952997208,
+    0.006762366741895676,
+    0.13754460215568542,
+    0.21149666607379913,
+    0.09019900858402252,
+    0.11703571677207947,
+    0.03510307893157005,
+    0.02578987367451191,
+    0.05109504982829094,
+    0.013695033267140388,
+    0.1303538978099823,
+    0.017273345962166786,
+    0.13908319175243378,
+    0.05268397554755211,
+    0.010254071094095707,
+    0.023861467838287354,
+    0.1337197870016098,
+    0.012882210314273834,
+    0.060103800147771835,
+    0.01640552468597889,
+    0.07166943699121475,
+    0.04852066561579704,
+    0.21064600348472595,
+    0.05943223834037781,
+    0.15058358013629913,
+    0.0426676981151104,
+    0.15856145322322845,
+    0.05052172765135765,
+    0.03798387572169304,
+    0.35575082898139954,
+    0.47427594661712646,
+    0.2916071116924286,
+    0.1713540256023407,
+    0.04815283790230751,
+    0.07065630704164505,
+    0.10667898505926132,
+    0.30502960085868835,
+    0.1437770277261734,
+    0.07106481492519379,
+    0.07805083692073822,
+    0.017001952975988388,
+    0.14516904950141907,
+    0.0064383684657514095,
+    0.30294954776763916,
+    0.10400618612766266,
+    0.09236416220664978,
+    0.22168096899986267,
+    0.1850627213716507,
+    0.09463033080101013,
+    0.02254527248442173,
+    0.10260853171348572,
+    0.06945618987083435,
+    0.022840889170765877,
+    0.0066554551012814045,
+    0.08745332807302475,
+    0.081999771296978,
+    0.0640682503581047,
+    0.23187373578548431,
+    0.32727351784706116,
+    0.12422331422567368,
+    0.15313737094402313,
+    0.20263177156448364,
+    0.07322326302528381,
+    0.055376723408699036,
+    0.11119652539491653,
+    0.10174702852964401,
+    0.12256954610347748,
+    0.10710106045007706,
+    0.3170746862888336,
+    0.012013965286314487,
+    0.1434120237827301,
+    0.02803770825266838,
+    0.01826053112745285,
+    0.10133804380893707,
+    0.019345903769135475,
+    0.01981542631983757,
+    0.044192396104335785,
+    0.02813100256025791,
+    0.17840521037578583,
+    0.07323722541332245,
+    0.010063420049846172,
+    0.1409100443124771,
+    0.2684270441532135,
+    0.09373287856578827,
+    0.3315849304199219,
+    0.04184216633439064,
+    0.05957753583788872,
+    0.057261306792497635,
+    0.2353627234697342,
+    0.1788700520992279,
+    0.05907813459634781,
+    0.010077272541821003,
+    0.12462282180786133,
+    0.11064265668392181
+  ]
+}

p4rel-mae-15m-d70-s0/mixture_plan.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2bd63859c1aff08f222ca30a4a99cd3315c12e2aa6ada89439b276fe8d628199
+size 15360128

p4rel-mae-15m-d70-s0/run.json

@@ -0,0 +1,94 @@
+{
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d70-s0",
+  "fingerprint": "06edcd5f801cd692",
+  "wandb_run_id": "phase4-reliability-p4rel-mae-15m-d70-s0",
+  "environment": {
+    "git_sha": "c3ed4f7086fd770b27c6d2c9980eb6b06ad4a3da",
+    "git_dirty": true,
+    "torch": "2.12.0+cu130",
+    "cuda": "13.0",
+    "cudnn": 92000,
+    "gpu_count": 1,
+    "gpu_name": "NVIDIA GeForce RTX 4090",
+    "gpu_memory_gib": 23.52,
+    "capability": "8.9",
+    "driver": "580.173.02"
+  },
+  "parameters": {
+    "encoder_params": 14294784,
+    "decoder_params": 647808,
+    "trainable_params": 15041152,
+    "total_params": 15041152
+  },
+  "schedule": {
+    "budget_clips": 1920000,
+    "schedule_clips": null,
+    "clips_per_step": 48,
+    "total_steps": 40000,
+    "schedule_steps": 40000,
+    "warmup_steps": 4000,
+    "ema_anneal_steps": 40000,
+    "final_lr": 6.000005700153392e-07,
+    "final_ema_decay": 0.9999899952500001
+  },
+  "geometry": {
+    "patch_grid": [
+      64,
+      4
+    ],
+    "num_patches": 256,
+    "visible_patches": 51,
+    "centers_per_mask": 2
+  },
+  "data": {
+    "split": "train",
+    "rows": 1912024,
+    "shards": 96,
+    "sampled_sha256": "58cddcb9592605014dfe6c40579aa65d2a3216cdb77feeaf0267c183e9f0d5a9",
+    "val_split": "balanced_train",
+    "val_clips": 2048
+  },
+  "mixture": {
+    "dist_id": null,
+    "has_weights": true,
+    "repetition_cap": 1000.0,
+    "distinct_fraction": null,
+    "max_repetition": 5.832684450853161,
+    "distinct_clips": 686706,
+    "capped_clusters": 0,
+    "realized_weights": [
+      0.004603645833333334,
+      0.058050520833333334,
+      0.0,
+      0.25281041666666665,
+      0.062025520833333334,
+      0.0,
+      0.16547291666666666,
+      0.0,
+      0.0021973958333333333,
+      0.0,
+      0.0074255208333333335,
+      0.016421875,
+      0.0008885416666666667,
+      0.0004182291666666667,
+      0.14711510416666668,
+      0.0,
+      0.24878489583333332,
+      0.0027208333333333334,
+      0.026671875,
+      0.0043927083333333335
+    ]
+  },
+  "started_at": 1786973255.7248852,
+  "finished_at": 1786973911.1545649,
+  "summary": {
+    "completed": true,
+    "final_step": 40000,
+    "clips_seen": 1920000,
+    "peak_gib": 0.6747117042541504,
+    "val/loss": 0.6318991957232356,
+    "val/frame": 0.6318991957232356,
+    "val/utterance": 0.0
+  }
+}

p4rel-mae-15m-d70-s0/status.json

@@ -0,0 +1,13 @@
+{
+  "state": "complete",
+  "updated_at": 1786973911.1558175,
+  "step": 40000,
+  "total_steps": 40000,
+  "clips_seen": 1920000,
+  "completed": true,
+  "final_step": 40000,
+  "peak_gib": 0.6747117042541504,
+  "val/loss": 0.6318991957232356,
+  "val/frame": 0.6318991957232356,
+  "val/utterance": 0.0
+}