Upload folder using huggingface_hub

p4rel-mae-15m-d40-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.02041015625
+  - 0.0
+  - 0.0
+  - 0.0
+  - 0.015484375
+  - 0.00492578125
+  - 0.00351953125
+  - 0.02041015625
+  - 0.05278515625
+  - 0.0
+  - 0.0
+  - 0.0
+  - 0.4331015625
+  - 0.0
+  - 0.0
+  - 0.4064296875
+  - 0.001408203125
+  - 0.0
+  - 0.041525390625
+  - 0.0
+  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-d40-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-d40-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.02041015625
+  - 0.0
+  - 0.0
+  - 0.0
+  - 0.015484375
+  - 0.00492578125
+  - 0.00351953125
+  - 0.02041015625
+  - 0.05278515625
+  - 0.0
+  - 0.0
+  - 0.0
+  - 0.4331015625
+  - 0.0
+  - 0.0
+  - 0.4064296875
+  - 0.001408203125
+  - 0.0
+  - 0.041525390625
+  - 0.0
+  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-d40-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-d40-s0/exports/step_00040000/fsd50k_lora_1ep_s0.json

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-15m-d40-s0/exports/step_00040000",
+  "fingerprint": "8398faff4cbadccd",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d40-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.3853212960064411,
+  "lora/map": 0.3853212960064411,
+  "fsd50k/map": 0.3853212960064411,
+  "train_seconds": 24.97,
+  "total_seconds": 34.25,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.023394437573151662,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.023565361441724275,
+      "val_map": 0.3853212960064411,
+      "lr": 0.0,
+      "train_s": 24.97211490373593
+    }
+  ],
+  "per_class_ap": [
+    0.28739336133003235,
+    0.06470432877540588,
+    0.5525945425033569,
+    0.37915289402008057,
+    0.4795542061328888,
+    0.6270549893379211,
+    0.8345900177955627,
+    0.5331564545631409,
+    0.5750936269760132,
+    0.4866996109485626,
+    0.2595159113407135,
+    0.37285009026527405,
+    0.1756991595029831,
+    0.40390464663505554,
+    0.5050023198127747,
+    0.47661781311035156,
+    0.025400884449481964,
+    0.2690572440624237,
+    0.459450900554657,
+    0.2193489521741867,
+    0.4746044874191284,
+    0.4136151671409607,
+    0.7798420190811157,
+    0.08003802597522736,
+    0.22319111227989197,
+    0.43003061413764954,
+    0.36337143182754517,
+    0.24764910340309143,
+    0.8073567748069763,
+    0.3278294503688812,
+    0.6613326668739319,
+    0.5557122230529785,
+    0.39863133430480957,
+    0.1712612807750702,
+    0.34308356046676636,
+    0.2194979041814804,
+    0.40745484828948975,
+    0.18678775429725647,
+    0.3328658640384674,
+    0.23637109994888306,
+    0.5135675668716431,
+    0.2615809738636017,
+    0.6531001925468445,
+    0.1573869287967682,
+    0.5074287056922913,
+    0.09280791878700256,
+    0.20909801125526428,
+    0.29524827003479004,
+    0.7090892195701599,
+    0.4000719487667084,
+    0.4620465040206909,
+    0.5625457167625427,
+    0.584029495716095,
+    0.15043234825134277,
+    0.1317988634109497,
+    0.03132358565926552,
+    0.30339816212654114,
+    0.6711106896400452,
+    0.1077054962515831,
+    0.5088849663734436,
+    0.5999535322189331,
+    0.6460513472557068,
+    0.4478387236595154,
+    0.16604559123516083,
+    0.5125017166137695,
+    0.13790935277938843,
+    0.2514069676399231,
+    0.6783756613731384,
+    0.5421435236930847,
+    0.5065243244171143,
+    0.49194270372390747,
+    0.27085554599761963,
+    0.6691566705703735,
+    0.2703176736831665,
+    0.27478769421577454,
+    0.21868835389614105,
+    0.33255159854888916,
+    0.3499140441417694,
+    0.2898824214935303,
+    0.4713522493839264,
+    0.2655220031738281,
+    0.2751677930355072,
+    0.3413573205471039,
+    0.36232107877731323,
+    0.24019333720207214,
+    0.2750403583049774,
+    0.4425697326660156,
+    0.5563174486160278,
+    0.26935040950775146,
+    0.14044345915317535,
+    0.7246933579444885,
+    0.12334299087524414,
+    0.5055307745933533,
+    0.20315374433994293,
+    0.3449469208717346,
+    0.31690001487731934,
+    0.5347923040390015,
+    0.12979234755039215,
+    0.3389751613140106,
+    0.09576787054538727,
+    0.562533974647522,
+    0.7034693956375122,
+    0.23129191994667053,
+    0.39634180068969727,
+    0.22767385840415955,
+    0.3930811583995819,
+    0.6838313937187195,
+    0.5283895134925842,
+    0.5352715849876404,
+    0.18162794411182404,
+    0.26234105229377747,
+    0.3347100615501404,
+    0.5104448795318604,
+    0.09319517016410828,
+    0.15127354860305786,
+    0.3335880637168884,
+    0.7467076778411865,
+    0.33792126178741455,
+    0.4883624017238617,
+    0.22491614520549774,
+    0.865200400352478,
+    0.8726983666419983,
+    0.3551172912120819,
+    0.13603536784648895,
+    0.24023336172103882,
+    0.7645444273948669,
+    0.28188252449035645,
+    0.7103537321090698,
+    0.318420946598053,
+    0.13359791040420532,
+    0.2190348356962204,
+    0.841565728187561,
+    0.42138922214508057,
+    0.4632094204425812,
+    0.6436580419540405,
+    0.09233644604682922,
+    0.08811339735984802,
+    0.05605899170041084,
+    0.5899607539176941,
+    0.554829478263855,
+    0.25334373116493225,
+    0.39368462562561035,
+    0.3134835660457611,
+    0.36726975440979004,
+    0.6096002459526062,
+    0.5435211062431335,
+    0.056046005338430405,
+    0.601108968257904,
+    0.16639158129692078,
+    0.2811546325683594,
+    0.44321945309638977,
+    0.5066966414451599,
+    0.25244686007499695,
+    0.3558788299560547,
+    0.5032936930656433,
+    0.32965564727783203,
+    0.4957225024700165,
+    0.6775295734405518,
+    0.4698237180709839,
+    0.06957916915416718,
+    0.40319305658340454,
+    0.14408300817012787,
+    0.6471328139305115,
+    0.2183399349451065,
+    0.33856239914894104,
+    0.34249719977378845,
+    0.34215518832206726,
+    0.06054113060235977,
+    0.36933109164237976,
+    0.4591065049171448,
+    0.10537592321634293,
+    0.8579698801040649,
+    0.8685778379440308,
+    0.07154687494039536,
+    0.529163658618927,
+    0.6990488767623901,
+    0.22488318383693695,
+    0.24703216552734375,
+    0.2239544540643692,
+    0.3014275133609772,
+    0.06216985359787941,
+    0.4220583140850067,
+    0.14091072976589203,
+    0.6230973601341248,
+    0.6146975159645081,
+    0.40609216690063477,
+    0.27640923857688904,
+    0.6586822271347046,
+    0.5911396145820618,
+    0.36916425824165344,
+    0.3486332297325134,
+    0.4209405183792114,
+    0.5675254464149475,
+    0.11469753831624985,
+    0.3214288353919983,
+    0.3206273317337036,
+    0.0999741181731224,
+    0.5306130647659302,
+    0.1255454272031784,
+    0.7568635940551758
+  ]
+}

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

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-15m-d40-s0/exports/step_00040000",
+  "fingerprint": "8398faff4cbadccd",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d40-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.38710991412401197,
+  "lora/map": 0.38710991412401197,
+  "fsd50k/map": 0.38710991412401197,
+  "train_seconds": 24.01,
+  "total_seconds": 33.7,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.023242766624316572,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.023413591722381843,
+      "val_map": 0.38710991412401197,
+      "lr": 0.0,
+      "train_s": 24.013270855299197
+    }
+  ],
+  "per_class_ap": [
+    0.26781463623046875,
+    0.08523088693618774,
+    0.48737528920173645,
+    0.3766424059867859,
+    0.48131728172302246,
+    0.6196973323822021,
+    0.8477761745452881,
+    0.5337882041931152,
+    0.5810530185699463,
+    0.4931079149246216,
+    0.28664955496788025,
+    0.39622846245765686,
+    0.16942009329795837,
+    0.3932512700557709,
+    0.49991297721862793,
+    0.4632200002670288,
+    0.025871943682432175,
+    0.29040804505348206,
+    0.4676697254180908,
+    0.2118062824010849,
+    0.48078376054763794,
+    0.37916359305381775,
+    0.8234211206436157,
+    0.08366598188877106,
+    0.26108676195144653,
+    0.3854095935821533,
+    0.37333640456199646,
+    0.247965469956398,
+    0.7766761183738708,
+    0.3039126396179199,
+    0.6527931094169617,
+    0.5140264630317688,
+    0.37423864006996155,
+    0.1647060066461563,
+    0.4215256869792938,
+    0.2347257286310196,
+    0.4268244504928589,
+    0.15177679061889648,
+    0.22280897200107574,
+    0.2843855321407318,
+    0.5550826787948608,
+    0.28511324524879456,
+    0.6381295919418335,
+    0.23393210768699646,
+    0.5670151114463806,
+    0.1471182107925415,
+    0.20093044638633728,
+    0.26289767026901245,
+    0.7160051465034485,
+    0.41647252440452576,
+    0.3293061852455139,
+    0.5085940361022949,
+    0.5591674447059631,
+    0.19432027637958527,
+    0.1303938776254654,
+    0.03670289367437363,
+    0.23153084516525269,
+    0.6645091772079468,
+    0.1260232925415039,
+    0.5292968153953552,
+    0.6059660315513611,
+    0.6484935879707336,
+    0.455620676279068,
+    0.09851919859647751,
+    0.5283724069595337,
+    0.12057354301214218,
+    0.33326196670532227,
+    0.6833611130714417,
+    0.5447778105735779,
+    0.5145055651664734,
+    0.48734331130981445,
+    0.19889263808727264,
+    0.6891688108444214,
+    0.33919668197631836,
+    0.24319691956043243,
+    0.23294185101985931,
+    0.4244987666606903,
+    0.38945725560188293,
+    0.30287864804267883,
+    0.42235067486763,
+    0.24455872178077698,
+    0.25334128737449646,
+    0.38932380080223083,
+    0.33959364891052246,
+    0.2580225169658661,
+    0.28604239225387573,
+    0.4184837341308594,
+    0.5482451319694519,
+    0.1786421686410904,
+    0.1463422179222107,
+    0.7445507645606995,
+    0.10468284040689468,
+    0.5252717733383179,
+    0.2508915960788727,
+    0.34661179780960083,
+    0.3637429177761078,
+    0.44242921471595764,
+    0.15689906477928162,
+    0.36657020449638367,
+    0.05992894619703293,
+    0.5680575370788574,
+    0.701328694820404,
+    0.226041778922081,
+    0.4021109938621521,
+    0.24822251498699188,
+    0.40444305539131165,
+    0.6871629953384399,
+    0.5099961161613464,
+    0.5437641143798828,
+    0.17610813677310944,
+    0.22173719108104706,
+    0.3301677703857422,
+    0.6169798970222473,
+    0.07909049838781357,
+    0.16242745518684387,
+    0.32201290130615234,
+    0.7139000296592712,
+    0.4382001757621765,
+    0.4959510862827301,
+    0.24539723992347717,
+    0.8716900944709778,
+    0.8801045417785645,
+    0.36447837948799133,
+    0.09342948347330093,
+    0.29459384083747864,
+    0.7712181806564331,
+    0.279727578163147,
+    0.732043981552124,
+    0.32064172625541687,
+    0.11942606419324875,
+    0.14227457344532013,
+    0.7901130318641663,
+    0.4138883054256439,
+    0.4623400568962097,
+    0.626743733882904,
+    0.08966761082410812,
+    0.07209724187850952,
+    0.049113065004348755,
+    0.5856214761734009,
+    0.5235874652862549,
+    0.21312811970710754,
+    0.33526989817619324,
+    0.3036474883556366,
+    0.30608683824539185,
+    0.6482664346694946,
+    0.5297998189926147,
+    0.06332699209451675,
+    0.526425838470459,
+    0.1944369077682495,
+    0.3412795662879944,
+    0.47265899181365967,
+    0.5170830488204956,
+    0.27515414357185364,
+    0.32190653681755066,
+    0.4701305329799652,
+    0.3881170451641083,
+    0.5600118637084961,
+    0.6896933913230896,
+    0.4598030745983124,
+    0.06420700252056122,
+    0.4066150188446045,
+    0.162798210978508,
+    0.6596831679344177,
+    0.24938195943832397,
+    0.28037750720977783,
+    0.3644827604293823,
+    0.3522442579269409,
+    0.06709027290344238,
+    0.36811158061027527,
+    0.3988819122314453,
+    0.12053388357162476,
+    0.861801028251648,
+    0.8729416728019714,
+    0.06017932668328285,
+    0.588947057723999,
+    0.7338494658470154,
+    0.22408513724803925,
+    0.2553573548793793,
+    0.20483556389808655,
+    0.3032362163066864,
+    0.034158915281295776,
+    0.47725629806518555,
+    0.17353485524654388,
+    0.6386932730674744,
+    0.6160663366317749,
+    0.40724417567253113,
+    0.2532147169113159,
+    0.6623817086219788,
+    0.6195700764656067,
+    0.3961630165576935,
+    0.4009678065776825,
+    0.39229217171669006,
+    0.5558972358703613,
+    0.11345183104276657,
+    0.35287266969680786,
+    0.39837709069252014,
+    0.12015768140554428,
+    0.5165424346923828,
+    0.18713749945163727,
+    0.7526674270629883
+  ]
+}

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

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-15m-d40-s0/exports/step_00040000",
+  "fingerprint": "8398faff4cbadccd",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d40-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.386523766303435,
+  "lora/map": 0.386523766303435,
+  "fsd50k/map": 0.386523766303435,
+  "train_seconds": 24.64,
+  "total_seconds": 33.69,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.02140844542882405,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.023689052770325895,
+      "val_map": 0.386523766303435,
+      "lr": 0.0,
+      "train_s": 24.638549690134823
+    }
+  ],
+  "per_class_ap": [
+    0.23181429505348206,
+    0.14751800894737244,
+    0.5273829698562622,
+    0.36070460081100464,
+    0.48239824175834656,
+    0.6173847317695618,
+    0.8224537372589111,
+    0.5219076871871948,
+    0.5669375061988831,
+    0.4331302046775818,
+    0.36719414591789246,
+    0.39033347368240356,
+    0.16849353909492493,
+    0.3439512848854065,
+    0.4975810945034027,
+    0.4680050015449524,
+    0.03881203010678291,
+    0.2261521816253662,
+    0.4200233221054077,
+    0.18227681517601013,
+    0.5831491947174072,
+    0.3866637051105499,
+    0.801540195941925,
+    0.09937570989131927,
+    0.25049224495887756,
+    0.3965705633163452,
+    0.3387240171432495,
+    0.25952228903770447,
+    0.7028020024299622,
+    0.3109878897666931,
+    0.6274600028991699,
+    0.5276585817337036,
+    0.3636040687561035,
+    0.1674291491508484,
+    0.36031365394592285,
+    0.23938266932964325,
+    0.41831904649734497,
+    0.18367892503738403,
+    0.4151524007320404,
+    0.24893221259117126,
+    0.5530728697776794,
+    0.21724054217338562,
+    0.6273002624511719,
+    0.17894160747528076,
+    0.5388445854187012,
+    0.19068820774555206,
+    0.3020823895931244,
+    0.2541145980358124,
+    0.6774012446403503,
+    0.4190114736557007,
+    0.3865838050842285,
+    0.4999372065067291,
+    0.5500036478042603,
+    0.12265387177467346,
+    0.18747320771217346,
+    0.024657418951392174,
+    0.20440828800201416,
+    0.6695384979248047,
+    0.11069183796644211,
+    0.48923784494400024,
+    0.5510660409927368,
+    0.6414236426353455,
+    0.45778319239616394,
+    0.22225379943847656,
+    0.4765818417072296,
+    0.152737095952034,
+    0.30088305473327637,
+    0.6525614857673645,
+    0.5400179028511047,
+    0.5280479788780212,
+    0.49752917885780334,
+    0.2877115309238434,
+    0.6880825161933899,
+    0.3247811198234558,
+    0.28098249435424805,
+    0.218079075217247,
+    0.47480055689811707,
+    0.395984947681427,
+    0.3291056752204895,
+    0.4191250801086426,
+    0.24050121009349823,
+    0.28839147090911865,
+    0.3713400065898895,
+    0.30340349674224854,
+    0.2801892161369324,
+    0.22055698931217194,
+    0.4072445034980774,
+    0.6349921226501465,
+    0.18087881803512573,
+    0.0787588581442833,
+    0.7351406216621399,
+    0.12111902981996536,
+    0.5754578113555908,
+    0.18845172226428986,
+    0.36356180906295776,
+    0.3734724819660187,
+    0.48828238248825073,
+    0.15992340445518494,
+    0.4176674783229828,
+    0.07886790484189987,
+    0.553830087184906,
+    0.7011871933937073,
+    0.22544367611408234,
+    0.424638569355011,
+    0.2487395703792572,
+    0.36440807580947876,
+    0.7048834562301636,
+    0.508766233921051,
+    0.5237484574317932,
+    0.21281227469444275,
+    0.18634384870529175,
+    0.3331180214881897,
+    0.596039891242981,
+    0.08489787578582764,
+    0.1439129114151001,
+    0.33609268069267273,
+    0.6509618759155273,
+    0.45465442538261414,
+    0.48427149653434753,
+    0.23112130165100098,
+    0.8746379613876343,
+    0.8823337554931641,
+    0.31322842836380005,
+    0.1236804947257042,
+    0.2877768874168396,
+    0.7668353319168091,
+    0.26362553238868713,
+    0.7267686724662781,
+    0.3342200815677643,
+    0.14915208518505096,
+    0.17413896322250366,
+    0.763143002986908,
+    0.49390336871147156,
+    0.45677614212036133,
+    0.6314387917518616,
+    0.1290026307106018,
+    0.0631435215473175,
+    0.057899072766304016,
+    0.650213360786438,
+    0.5169677734375,
+    0.1389724761247635,
+    0.3155309557914734,
+    0.23889882862567902,
+    0.36148160696029663,
+    0.6708497405052185,
+    0.5716634392738342,
+    0.052944861352443695,
+    0.5888412594795227,
+    0.16807103157043457,
+    0.3384817838668823,
+    0.46496549248695374,
+    0.4516431987285614,
+    0.35027000308036804,
+    0.3627970516681671,
+    0.45113900303840637,
+    0.33914420008659363,
+    0.5295158624649048,
+    0.6176320314407349,
+    0.45242050290107727,
+    0.068865567445755,
+    0.3804430067539215,
+    0.20936298370361328,
+    0.6409932374954224,
+    0.2793484628200531,
+    0.32087263464927673,
+    0.4703594148159027,
+    0.3567449152469635,
+    0.0639423131942749,
+    0.28177550435066223,
+    0.4517488479614258,
+    0.10517901182174683,
+    0.8559802770614624,
+    0.8643322587013245,
+    0.05456700921058655,
+    0.5370875000953674,
+    0.7071268558502197,
+    0.2449193149805069,
+    0.28096842765808105,
+    0.2052188366651535,
+    0.30964675545692444,
+    0.06553370505571365,
+    0.4880100190639496,
+    0.1488364040851593,
+    0.58885657787323,
+    0.6076492667198181,
+    0.35751354694366455,
+    0.27566802501678467,
+    0.6537463665008545,
+    0.5697595477104187,
+    0.3428483009338379,
+    0.46362966299057007,
+    0.3920905292034149,
+    0.5585684180259705,
+    0.13490411639213562,
+    0.29177454113960266,
+    0.3789577782154083,
+    0.116185262799263,
+    0.5004310011863708,
+    0.15696856379508972,
+    0.7814810276031494
+  ]
+}

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

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-15m-d40-s0/exports/step_00040000",
+    "fingerprint": "8398faff4cbadccd",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-15m-d40-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.0069142237410127795,
+      "val_map": 0.006050261742421258,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02356,
+      "train_map": 0.030231907276526814,
+      "val_map": null,
+      "lr": 0.0017857142857142857,
+      "train_s": 0.9
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.02325,
+      "train_map": 0.053521918617780946,
+      "val_map": null,
+      "lr": 0.002996832162669313,
+      "train_s": 1.5
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02246,
+      "train_map": 0.06576672426126261,
+      "val_map": null,
+      "lr": 0.0029464015850084926,
+      "train_s": 2.2
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02209,
+      "train_map": 0.07598944377919316,
+      "val_map": null,
+      "lr": 0.0028365019165406584,
+      "train_s": 2.8
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02205,
+      "train_map": 0.08887145466783546,
+      "val_map": 0.0536087313923186,
+      "lr": 0.002795633084492174,
+      "train_s": 3.0
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02105,
+      "train_map": 0.09526989015130294,
+      "val_map": null,
+      "lr": 0.0026716673782849554,
+      "train_s": 3.5
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02061,
+      "train_map": 0.09914778177122775,
+      "val_map": null,
+      "lr": 0.002458698684369641,
+      "train_s": 4.1
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.02041,
+      "train_map": 0.12524776762586434,
+      "val_map": null,
+      "lr": 0.0022063824593754194,
+      "train_s": 4.7
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.02028,
+      "train_map": 0.13882625573252874,
+      "val_map": null,
+      "lr": 0.0019251287213377403,
+      "train_s": 5.4
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.01965,
+      "train_map": 0.141823241213227,
+      "val_map": 0.0864240971006863,
+      "lr": 0.0017592766878384548,
+      "train_s": 5.8
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01994,
+      "train_map": 0.14990596904329234,
+      "val_map": null,
+      "lr": 0.0016265413870700269,
+      "train_s": 6.2
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.0196,
+      "train_map": 0.17398937804398235,
+      "val_map": null,
+      "lr": 0.0013229395198229876,
+      "train_s": 6.8
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01944,
+      "train_map": 0.18584892295735694,
+      "val_map": null,
+      "lr": 0.0010268490715591258,
+      "train_s": 7.5
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01907,
+      "train_map": 0.20183882758859015,
+      "val_map": null,
+      "lr": 0.000750486089449494,
+      "train_s": 8.1
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.0184,
+      "train_map": 0.21102153818411098,
+      "val_map": 0.12495186858487084,
+      "lr": 0.000541962785848462,
+      "train_s": 8.7
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01884,
+      "train_map": 0.21343386907791417,
+      "val_map": null,
+      "lr": 0.0005052527083673369,
+      "train_s": 8.8
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01808,
+      "train_map": 0.22180304283442318,
+      "val_map": null,
+      "lr": 0.00030126672363088163,
+      "train_s": 9.4
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01767,
+      "train_map": 0.234200963359927,
+      "val_map": null,
+      "lr": 0.00014694415279957929,
+      "train_s": 10.0
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01737,
+      "train_map": 0.2371027988365779,
+      "val_map": null,
+      "lr": 4.865200911511649e-05,
+      "train_s": 10.7
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.018,
+      "train_map": 0.2374037219038272,
+      "val_map": null,
+      "lr": 1.0445612398966596e-05,
+      "train_s": 11.3
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.01803,
+      "train_map": 0.23720514271726417,
+      "val_map": 0.13995561996506672,
+      "lr": 1e-05,
+      "train_s": 11.3
+    }
+  ],
+  "final_train_map": 0.23720514271726417,
+  "final_val_map": 0.13995561996506672,
+  "lora/map": 0.13995561996506672,
+  "train_seconds": 11.3,
+  "wall_seconds": 66.2
+}

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

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-15m-d40-s0/exports/step_00040000",
+    "fingerprint": "8398faff4cbadccd",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-15m-d40-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.006989543099885019,
+      "val_map": 0.0062052592488475835,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02457,
+      "train_map": 0.02881654834266985,
+      "val_map": null,
+      "lr": 0.0017857142857142857,
+      "train_s": 0.6
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.02309,
+      "train_map": 0.05826502305939723,
+      "val_map": null,
+      "lr": 0.002996832162669313,
+      "train_s": 1.2
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02232,
+      "train_map": 0.0642703775068183,
+      "val_map": null,
+      "lr": 0.0029464015850084926,
+      "train_s": 1.8
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02159,
+      "train_map": 0.07891468953542709,
+      "val_map": null,
+      "lr": 0.0028365019165406584,
+      "train_s": 2.4
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02193,
+      "train_map": 0.083496282143126,
+      "val_map": 0.05398396245889592,
+      "lr": 0.002795633084492174,
+      "train_s": 2.6
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02103,
+      "train_map": 0.0838732783703851,
+      "val_map": null,
+      "lr": 0.0026716673782849554,
+      "train_s": 3.0
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02107,
+      "train_map": 0.10491581544354275,
+      "val_map": null,
+      "lr": 0.002458698684369641,
+      "train_s": 3.6
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.02038,
+      "train_map": 0.11157267104102692,
+      "val_map": null,
+      "lr": 0.0022063824593754194,
+      "train_s": 4.2
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.02064,
+      "train_map": 0.12070453024347379,
+      "val_map": null,
+      "lr": 0.0019251287213377403,
+      "train_s": 4.8
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.01991,
+      "train_map": 0.14666314746661419,
+      "val_map": 0.08694938903469486,
+      "lr": 0.0017592766878384548,
+      "train_s": 5.1
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01989,
+      "train_map": 0.14735262705352462,
+      "val_map": null,
+      "lr": 0.0016265413870700269,
+      "train_s": 5.4
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.01935,
+      "train_map": 0.15286650481868883,
+      "val_map": null,
+      "lr": 0.0013229395198229876,
+      "train_s": 6.0
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.0192,
+      "train_map": 0.17109744316249717,
+      "val_map": null,
+      "lr": 0.0010268490715591258,
+      "train_s": 6.6
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.0188,
+      "train_map": 0.19371618901112556,
+      "val_map": null,
+      "lr": 0.000750486089449494,
+      "train_s": 7.2
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.01831,
+      "train_map": 0.20470569875759323,
+      "val_map": 0.12420615879288632,
+      "lr": 0.000541962785848462,
+      "train_s": 7.7
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01867,
+      "train_map": 0.20561892974572132,
+      "val_map": null,
+      "lr": 0.0005052527083673369,
+      "train_s": 7.7
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01789,
+      "train_map": 0.21593763312126954,
+      "val_map": null,
+      "lr": 0.00030126672363088163,
+      "train_s": 8.3
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01817,
+      "train_map": 0.2217271935426236,
+      "val_map": null,
+      "lr": 0.00014694415279957929,
+      "train_s": 8.9
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01786,
+      "train_map": 0.22568905362028777,
+      "val_map": null,
+      "lr": 4.865200911511649e-05,
+      "train_s": 9.5
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.01781,
+      "train_map": 0.2267279289768871,
+      "val_map": null,
+      "lr": 1.0445612398966596e-05,
+      "train_s": 10.1
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.01757,
+      "train_map": 0.22661605559155146,
+      "val_map": 0.13783237366076467,
+      "lr": 1e-05,
+      "train_s": 10.2
+    }
+  ],
+  "final_train_map": 0.22661605559155146,
+  "final_val_map": 0.13783237366076467,
+  "lora/map": 0.13783237366076467,
+  "train_seconds": 10.2,
+  "wall_seconds": 61.8
+}

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

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-15m-d40-s0/exports/step_00040000",
+    "fingerprint": "8398faff4cbadccd",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-15m-d40-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.006592422163673275,
+      "val_map": 0.00603006282565935,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02419,
+      "train_map": 0.03163473994229908,
+      "val_map": null,
+      "lr": 0.0017857142857142857,
+      "train_s": 0.6
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.02351,
+      "train_map": 0.04030145202547778,
+      "val_map": null,
+      "lr": 0.002996832162669313,
+      "train_s": 1.2
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02237,
+      "train_map": 0.06308820212866927,
+      "val_map": null,
+      "lr": 0.0029464015850084926,
+      "train_s": 1.8
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02149,
+      "train_map": 0.06990912788197667,
+      "val_map": null,
+      "lr": 0.0028365019165406584,
+      "train_s": 2.4
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02196,
+      "train_map": 0.07841617879157345,
+      "val_map": 0.049965261001461515,
+      "lr": 0.002795633084492174,
+      "train_s": 2.6
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02156,
+      "train_map": 0.08994236260261987,
+      "val_map": null,
+      "lr": 0.0026716673782849554,
+      "train_s": 3.0
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.0212,
+      "train_map": 0.10569097748577623,
+      "val_map": null,
+      "lr": 0.002458698684369641,
+      "train_s": 3.6
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.02089,
+      "train_map": 0.11700108890736254,
+      "val_map": null,
+      "lr": 0.0022063824593754194,
+      "train_s": 4.2
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.01981,
+      "train_map": 0.12402514291598517,
+      "val_map": null,
+      "lr": 0.0019251287213377403,
+      "train_s": 4.7
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.0193,
+      "train_map": 0.13283790248239402,
+      "val_map": 0.08346765110732199,
+      "lr": 0.0017592766878384548,
+      "train_s": 5.1
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01929,
+      "train_map": 0.1412338162401242,
+      "val_map": null,
+      "lr": 0.0016265413870700269,
+      "train_s": 5.4
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.0191,
+      "train_map": 0.16033647016691047,
+      "val_map": null,
+      "lr": 0.0013229395198229876,
+      "train_s": 6.0
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01926,
+      "train_map": 0.17451235328657833,
+      "val_map": null,
+      "lr": 0.0010268490715591258,
+      "train_s": 6.6
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01858,
+      "train_map": 0.18910327912405472,
+      "val_map": null,
+      "lr": 0.000750486089449494,
+      "train_s": 7.1
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.0184,
+      "train_map": 0.20184308080536364,
+      "val_map": 0.122270835350591,
+      "lr": 0.000541962785848462,
+      "train_s": 7.6
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01816,
+      "train_map": 0.20129651129616874,
+      "val_map": null,
+      "lr": 0.0005052527083673369,
+      "train_s": 7.7
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01825,
+      "train_map": 0.21672973897134923,
+      "val_map": null,
+      "lr": 0.00030126672363088163,
+      "train_s": 8.3
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01819,
+      "train_map": 0.2228143013266235,
+      "val_map": null,
+      "lr": 0.00014694415279957929,
+      "train_s": 8.9
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01786,
+      "train_map": 0.22495814916046877,
+      "val_map": null,
+      "lr": 4.865200911511649e-05,
+      "train_s": 9.5
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.018,
+      "train_map": 0.22627832635042508,
+      "val_map": null,
+      "lr": 1.0445612398966596e-05,
+      "train_s": 10.0
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.0183,
+      "train_map": 0.22643904942061321,
+      "val_map": 0.13695873313475634,
+      "lr": 1e-05,
+      "train_s": 10.1
+    }
+  ],
+  "final_train_map": 0.22643904942061321,
+  "final_val_map": 0.13695873313475634,
+  "lora/map": 0.13695873313475634,
+  "train_seconds": 10.1,
+  "wall_seconds": 63.8
+}

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

@@ -0,0 +1,576 @@
+{
+  "export_dir": "/workspace/runs/p4rel-mae-15m-d40-s0/exports/step_00040000",
+  "fingerprint": "8398faff4cbadccd",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d40-s0",
+  "probe/map": 0.14903043944784483,
+  "probe/map_train": 0.5178566184283661,
+  "probe/alpha": 1.0,
+  "probe/holdout_map": 0.1725976581807183,
+  "probe/pooling": "concat",
+  "probe/feature_dim": 768,
+  "probe/fit_clips": 20550,
+  "probe/score_clips": 18886,
+  "probe/classes_scored": 527,
+  "probe/extract_s": 6.890961027005687,
+  "probe/total_s": 8.367473515041638,
+  "alpha_selection": [
+    {
+      "alpha": 0.01,
+      "holdout_map": 0.17078042185098372
+    },
+    {
+      "alpha": 0.1,
+      "holdout_map": 0.1711911997462727
+    },
+    {
+      "alpha": 1.0,
+      "holdout_map": 0.1725976581807183
+    },
+    {
+      "alpha": 10.0,
+      "holdout_map": 0.17006748613143302
+    },
+    {
+      "alpha": 100.0,
+      "holdout_map": 0.15891913006009722
+    },
+    {
+      "alpha": 1000.0,
+      "holdout_map": 0.13635189771480802
+    },
+    {
+      "alpha": 10000.0,
+      "holdout_map": 0.1086260058415216
+    }
+  ],
+  "per_class_ap": [
+    0.6365665197372437,
+    0.009304622188210487,
+    0.06327573210000992,
+    0.198362335562706,
+    0.11474429070949554,
+    0.27195295691490173,
+    0.19500911235809326,
+    0.3164454996585846,
+    0.04555390402674675,
+    0.050201792269945145,
+    0.026514850556850433,
+    0.03857136517763138,
+    0.5027457475662231,
+    0.32655373215675354,
+    0.05811311677098274,
+    0.5336174964904785,
+    0.34686681628227234,
+    0.3715859353542328,
+    0.06900274753570557,
+    0.13058926165103912,
+    0.2464790940284729,
+    0.11583314836025238,
+    0.21575835347175598,
+    0.15960025787353516,
+    0.08476429432630539,
+    0.029251759871840477,
+    0.08222516626119614,
+    0.16879421472549438,
+    0.2273816615343094,
+    0.11398189514875412,
+    0.23678652942180634,
+    0.1311405450105667,
+    0.019314618781208992,
+    0.06970488280057907,
+    0.2688937485218048,
+    0.04994131624698639,
+    0.3920336663722992,
+    0.06675896048545837,
+    0.0818377286195755,
+    0.16795021295547485,
+    0.17197845876216888,
+    0.03237423673272133,
+    0.08029136061668396,
+    0.4344981014728546,
+    0.016962507739663124,
+    0.09124783426523209,
+    0.01811395399272442,
+    0.14389291405677795,
+    0.05188180133700371,
+    0.07236819714307785,
+    0.11754090338945389,
+    0.011986668221652508,
+    0.04183712601661682,
+    0.039426352828741074,
+    0.2307530790567398,
+    0.11873467266559601,
+    0.2460116595029831,
+    0.35593676567077637,
+    0.16046598553657532,
+    0.0788545161485672,
+    0.19651678204536438,
+    0.22568635642528534,
+    0.05994356796145439,
+    0.161885067820549,
+    0.5430777072906494,
+    0.255369633436203,
+    0.29526078701019287,
+    0.3253193795681,
+    0.41280490159988403,
+    0.34346893429756165,
+    0.09928160905838013,
+    0.1295115351676941,
+    0.20777860283851624,
+    0.1988905668258667,
+    0.1823570430278778,
+    0.20383499562740326,
+    0.04653092473745346,
+    0.23371152579784393,
+    0.1692611128091812,
+    0.15805023908615112,
+    0.021337874233722687,
+    0.12455815821886063,
+    0.37442463636398315,
+    0.15177828073501587,
+    0.10759265720844269,
+    0.03959164395928383,
+    0.06436386704444885,
+    0.0422300286591053,
+    0.03680143877863884,
+    0.06904049217700958,
+    0.07052741199731827,
+    0.06062057614326477,
+    0.2163693755865097,
+    0.02623538114130497,
+    0.03821679204702377,
+    0.04620078578591347,
+    0.03558208420872688,
+    0.025538813322782516,
+    0.1410277932882309,
+    0.23679839074611664,
+    0.2262955605983734,
+    0.3498115539550781,
+    0.04733380302786827,
+    0.047509465366601944,
+    0.01743183098733425,
+    0.03439661115407944,
+    0.06906998157501221,
+    0.057738158851861954,
+    0.09462986141443253,
+    0.1827661246061325,
+    0.2098374217748642,
+    0.1418793797492981,
+    0.12572893500328064,
+    0.1009075865149498,
+    0.04883817955851555,
+    0.2547864615917206,
+    0.25503548979759216,
+    0.1007000207901001,
+    0.12592770159244537,
+    0.2051275670528412,
+    0.2161020189523697,
+    0.17465032637119293,
+    0.09470421075820923,
+    0.09634269773960114,
+    0.008884061127901077,
+    0.03898059204220772,
+    0.12563331425189972,
+    0.17721351981163025,
+    0.11764630675315857,
+    0.12399262934923172,
+    0.005945994984358549,
+    0.13440120220184326,
+    0.12527996301651,
+    0.11753895878791809,
+    0.06797516345977783,
+    0.010266593657433987,
+    0.10111302882432938,
+    0.724972665309906,
+    0.24927499890327454,
+    0.3054187297821045,
+    0.40780094265937805,
+    0.30951592326164246,
+    0.19079209864139557,
+    0.14637407660484314,
+    0.07683753967285156,
+    0.12518933415412903,
+    0.08560514450073242,
+    0.24618814885616302,
+    0.16651007533073425,
+    0.2052604705095291,
+    0.15982988476753235,
+    0.18744686245918274,
+    0.27344879508018494,
+    0.12316858768463135,
+    0.10857410728931427,
+    0.18987983465194702,
+    0.07299372553825378,
+    0.07344544678926468,
+    0.030373480170965195,
+    0.07589942961931229,
+    0.48182186484336853,
+    0.32213497161865234,
+    0.3513108789920807,
+    0.16648449003696442,
+    0.2839430272579193,
+    0.3244404196739197,
+    0.3342670202255249,
+    0.20895959436893463,
+    0.32308509945869446,
+    0.3544095456600189,
+    0.2168317288160324,
+    0.3253145217895508,
+    0.3713051676750183,
+    0.16401606798171997,
+    0.08611195534467697,
+    0.08623318374156952,
+    0.12449593096971512,
+    0.3461681306362152,
+    0.07608038932085037,
+    0.18370965123176575,
+    0.2164846956729889,
+    0.1980488896369934,
+    0.16098017990589142,
+    0.2834501564502716,
+    0.08637668937444687,
+    0.4020218849182129,
+    0.3258846402168274,
+    0.2004079818725586,
+    0.2747476100921631,
+    0.16564767062664032,
+    0.16762188076972961,
+    0.10177253931760788,
+    0.3363770544528961,
+    0.18980294466018677,
+    0.18801623582839966,
+    0.3822365999221802,
+    0.06721633672714233,
+    0.11267130076885223,
+    0.09119844436645508,
+    0.27161070704460144,
+    0.07203120738267899,
+    0.24159777164459229,
+    0.14224593341350555,
+    0.23048993945121765,
+    0.20863185822963715,
+    0.2173900306224823,
+    0.24492762982845306,
+    0.5385814309120178,
+    0.21995000541210175,
+    0.324148565530777,
+    0.7529703378677368,
+    0.42233216762542725,
+    0.501499593257904,
+    0.048681821674108505,
+    0.33916887640953064,
+    0.14490556716918945,
+    0.07084721326828003,
+    0.25773805379867554,
+    0.17397448420524597,
+    0.14245550334453583,
+    0.18788833916187286,
+    0.1445409208536148,
+    0.20231448113918304,
+    0.2587229609489441,
+    0.051886070519685745,
+    0.08238592743873596,
+    0.06357729434967041,
+    0.033892951905727386,
+    0.1372387558221817,
+    0.05728595331311226,
+    0.04716376215219498,
+    0.04871169850230217,
+    0.0705433040857315,
+    0.042184848338365555,
+    0.07122435420751572,
+    0.013069741427898407,
+    0.1855769008398056,
+    0.02194335125386715,
+    0.3080352246761322,
+    0.029883574694395065,
+    0.24186404049396515,
+    0.2671206593513489,
+    0.3277159035205841,
+    0.21877998113632202,
+    0.14825530350208282,
+    0.23570343852043152,
+    0.07005840539932251,
+    0.22937177121639252,
+    0.026336554437875748,
+    0.12967275083065033,
+    0.08325114846229553,
+    0.021468278020620346,
+    0.06488700211048126,
+    0.08928157389163971,
+    0.0559881255030632,
+    0.21126289665699005,
+    0.2374645471572876,
+    0.21603825688362122,
+    0.06874227523803711,
+    0.07274229824542999,
+    0.05127589404582977,
+    0.08454225957393646,
+    0.11874499171972275,
+    0.06113098934292793,
+    0.02152225375175476,
+    0.05247516557574272,
+    0.06354214251041412,
+    0.10104357451200485,
+    0.061048638075590134,
+    0.05186004936695099,
+    0.02679719775915146,
+    0.24414825439453125,
+    0.14584867656230927,
+    0.034661658108234406,
+    0.19222287833690643,
+    0.034389156848192215,
+    0.0380207821726799,
+    0.02210516668856144,
+    0.07968404144048691,
+    0.07234549522399902,
+    0.02781326323747635,
+    0.26145949959754944,
+    0.14857302606105804,
+    0.2569029927253723,
+    0.06116176024079323,
+    0.28615447878837585,
+    0.3519262969493866,
+    0.3561662435531616,
+    0.260476291179657,
+    0.22700488567352295,
+    0.2129541039466858,
+    0.2009650468826294,
+    0.10682780295610428,
+    0.2062128484249115,
+    0.22625385224819183,
+    0.24521833658218384,
+    0.14121609926223755,
+    0.007789468392729759,
+    0.03475125879049301,
+    0.01857713982462883,
+    0.2955434322357178,
+    0.22171618044376373,
+    0.20624321699142456,
+    0.1012074276804924,
+    0.02404501847922802,
+    0.10848425328731537,
+    0.04277922585606575,
+    0.19236232340335846,
+    0.04850394278764725,
+    0.09832415729761124,
+    0.17804208397865295,
+    0.0791655108332634,
+    0.15651766955852509,
+    0.08308272808790207,
+    0.09015725553035736,
+    0.15338511765003204,
+    0.06165786460042,
+    0.08273646235466003,
+    0.1615743488073349,
+    0.06050002947449684,
+    0.25118690729141235,
+    0.03811926394701004,
+    0.4441932439804077,
+    0.3719148635864258,
+    0.268414705991745,
+    0.26780128479003906,
+    0.07035152614116669,
+    0.07132507115602493,
+    0.3049802780151367,
+    0.2900646924972534,
+    0.1487305462360382,
+    0.24823710322380066,
+    0.31209033727645874,
+    0.3820267915725708,
+    0.2567928731441498,
+    0.20527520775794983,
+    0.21584568917751312,
+    0.2969847619533539,
+    0.05577525869011879,
+    0.03398328647017479,
+    0.15559916198253632,
+    0.009374963119626045,
+    0.05951788276433945,
+    0.1794571429491043,
+    0.10721329599618912,
+    0.11190887540578842,
+    0.11779868602752686,
+    0.36424973607063293,
+    0.0299635361880064,
+    0.020982922986149788,
+    0.14981205761432648,
+    0.026005811989307404,
+    0.09719197452068329,
+    0.09607540816068649,
+    0.02803998813033104,
+    0.042584214359521866,
+    0.02692047506570816,
+    0.022973595187067986,
+    0.03526471555233002,
+    0.1350155770778656,
+    0.02055896259844303,
+    0.06492059677839279,
+    0.14992059767246246,
+    0.12166574597358704,
+    0.039917875081300735,
+    0.05142645910382271,
+    0.354751318693161,
+    0.2022332102060318,
+    0.020526114851236343,
+    0.1465500146150589,
+    0.15652704238891602,
+    0.12945865094661713,
+    0.1409561187028885,
+    0.3996521532535553,
+    0.16588924825191498,
+    0.09776175767183304,
+    0.09182462841272354,
+    0.10593058913946152,
+    0.021314533427357674,
+    0.020068814978003502,
+    0.04634387791156769,
+    0.1152266189455986,
+    0.1975051313638687,
+    0.08029700070619583,
+    0.19766898453235626,
+    0.1857794225215912,
+    0.1926628202199936,
+    0.03559340909123421,
+    0.06743897497653961,
+    0.06810245662927628,
+    0.26364922523498535,
+    0.10784835368394852,
+    0.038756776601076126,
+    0.38096827268600464,
+    0.3057665228843689,
+    0.09416189044713974,
+    0.49243226647377014,
+    0.584780752658844,
+    0.02634456753730774,
+    0.39212122559547424,
+    0.43057000637054443,
+    0.050815850496292114,
+    0.1393476277589798,
+    0.29696348309516907,
+    0.03360338881611824,
+    0.04511921480298042,
+    0.061146609485149384,
+    0.08333630859851837,
+    0.1027870699763298,
+    0.054180826991796494,
+    0.021960502490401268,
+    0.09952805191278458,
+    0.025157717987895012,
+    0.04975675791501999,
+    0.29196682572364807,
+    0.06567047536373138,
+    0.028375238180160522,
+    0.015963928773999214,
+    0.11423054337501526,
+    0.12267445772886276,
+    0.2859242856502533,
+    0.07385872304439545,
+    0.07976625114679337,
+    0.23698313534259796,
+    0.10058863461017609,
+    0.09888538718223572,
+    0.05347612500190735,
+    0.18891681730747223,
+    0.177018940448761,
+    0.20059899985790253,
+    0.07820150256156921,
+    0.1696627140045166,
+    0.22480231523513794,
+    0.12800170481204987,
+    0.005701479502022266,
+    0.0986441895365715,
+    0.22404395043849945,
+    0.08939020335674286,
+    0.08705870807170868,
+    0.0502747967839241,
+    0.027408242225646973,
+    0.04562034085392952,
+    0.028565119951963425,
+    0.0981757640838623,
+    0.018347669392824173,
+    0.10885552316904068,
+    0.03461584821343422,
+    0.009498697705566883,
+    0.02482139877974987,
+    0.10321184992790222,
+    0.010296612046658993,
+    0.0590062215924263,
+    0.016228439286351204,
+    0.04873267933726311,
+    0.05173369497060776,
+    0.20596425235271454,
+    0.050791822373867035,
+    0.13466303050518036,
+    0.04163168743252754,
+    0.12172949314117432,
+    0.06407889723777771,
+    0.02918248251080513,
+    0.36188963055610657,
+    0.4562952220439911,
+    0.22794106602668762,
+    0.1460026204586029,
+    0.034198351204395294,
+    0.06572026759386063,
+    0.04932970181107521,
+    0.18580053746700287,
+    0.08630968630313873,
+    0.06257601827383041,
+    0.07461579144001007,
+    0.020273922011256218,
+    0.09405917674303055,
+    0.005776406731456518,
+    0.28084710240364075,
+    0.06328430026769638,
+    0.09742356836795807,
+    0.2717602252960205,
+    0.16158446669578552,
+    0.09517531096935272,
+    0.020249230787158012,
+    0.10599207133054733,
+    0.0655336007475853,
+    0.01457769051194191,
+    0.009597807191312313,
+    0.07192016392946243,
+    0.04867270961403847,
+    0.04606013372540474,
+    0.214818075299263,
+    0.2879859507083893,
+    0.10144814848899841,
+    0.145217627286911,
+    0.22262075543403625,
+    0.03736693784594536,
+    0.04599546641111374,
+    0.06724580377340317,
+    0.05159320682287216,
+    0.11386509984731674,
+    0.10619921237230301,
+    0.3449631631374359,
+    0.006857526488602161,
+    0.17218761146068573,
+    0.03021566942334175,
+    0.025102470070123672,
+    0.10284626483917236,
+    0.018224168568849564,
+    0.0206650011241436,
+    0.04197590798139572,
+    0.02210311032831669,
+    0.11980608105659485,
+    0.0694657489657402,
+    0.008581151254475117,
+    0.14322277903556824,
+    0.2776675224304199,
+    0.0703033059835434,
+    0.3258082866668701,
+    0.04341991990804672,
+    0.03280526399612427,
+    0.05093766003847122,
+    0.2161431610584259,
+    0.17063497006893158,
+    0.07833016663789749,
+    0.01160561852157116,
+    0.11095231026411057,
+    0.10498224198818207
+  ]
+}

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

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

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

@@ -0,0 +1,94 @@
+{
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-15m-d40-s0",
+  "fingerprint": "8398faff4cbadccd",
+  "wandb_run_id": "phase4-reliability-p4rel-mae-15m-d40-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": 7.5,
+    "distinct_clips": 485516,
+    "capped_clusters": 0,
+    "realized_weights": [
+      0.020410416666666667,
+      0.0,
+      0.0,
+      0.0,
+      0.015484375,
+      0.004925520833333333,
+      0.0035192708333333335,
+      0.020410416666666667,
+      0.05278489583333333,
+      0.0,
+      0.0,
+      0.0,
+      0.4331015625,
+      0.0,
+      0.0,
+      0.4064296875,
+      0.0014083333333333333,
+      0.0,
+      0.041525520833333336,
+      0.0
+    ]
+  },
+  "started_at": 1786973933.6887693,
+  "finished_at": 1786974590.6709464,
+  "summary": {
+    "completed": true,
+    "final_step": 40000,
+    "clips_seen": 1920000,
+    "peak_gib": 0.6747117042541504,
+    "val/loss": 0.6333574545569718,
+    "val/frame": 0.6333574545569718,
+    "val/utterance": 0.0
+  }
+}

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

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