Upload folder using huggingface_hub

p4rel-mae-base-d40-s1/config.yaml

@@ -0,0 +1,112 @@
+model:
+  embed_dim: 768
+  depth: 12
+  num_heads: 12
+  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: 24
+  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.0001
+  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: 2
+  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-base-d40-s1
+  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: 1
+  data_loader_seed: null
+  objective_seed: null
+  deterministic: false
+  allow_tf32: true

p4rel-mae-base-d40-s1/exports/step_00040000/config.yaml

@@ -0,0 +1,112 @@
+model:
+  embed_dim: 768
+  depth: 12
+  num_heads: 12
+  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: 24
+  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.0001
+  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: 2
+  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-base-d40-s1
+  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: 1
+  data_loader_seed: null
+  objective_seed: null
+  deterministic: false
+  allow_tf32: true

p4rel-mae-base-d40-s1/exports/step_00040000/fsd50k_lora_1ep_s0.json

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-base-d40-s1/exports/step_00040000",
+  "fingerprint": "bbcd70148f3424d1",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-base-d40-s1",
+  "target": "fsd50k",
+  "num_classes": 200,
+  "lora_layers_wrapped": 48,
+  "params": {
+    "trainable": 2551496,
+    "total": 87767240,
+    "fraction": 0.02907116596124021
+  },
+  "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.001,
+    "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.3993412546440959,
+  "lora/map": 0.3993412546440959,
+  "fsd50k/map": 0.3993412546440959,
+  "train_seconds": 88.17,
+  "total_seconds": 116.57,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.022526441204827278,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.02265060402000164,
+      "val_map": 0.3993412546440959,
+      "lr": 0.0,
+      "train_s": 88.17066002381034
+    }
+  ],
+  "per_class_ap": [
+    0.23334603011608124,
+    0.11626867204904556,
+    0.5253791809082031,
+    0.3729081451892853,
+    0.485107421875,
+    0.6354355812072754,
+    0.845313310623169,
+    0.5772477984428406,
+    0.6041590571403503,
+    0.45306411385536194,
+    0.3477664589881897,
+    0.4233602285385132,
+    0.29687511920928955,
+    0.4807497560977936,
+    0.5165684223175049,
+    0.48060983419418335,
+    0.03143709897994995,
+    0.25326067209243774,
+    0.4599090814590454,
+    0.1913318932056427,
+    0.48718589544296265,
+    0.390787273645401,
+    0.8569830060005188,
+    0.10465876013040543,
+    0.2483442723751068,
+    0.4150933623313904,
+    0.3570096492767334,
+    0.28071704506874084,
+    0.7578405737876892,
+    0.33814406394958496,
+    0.6499671339988708,
+    0.6244993805885315,
+    0.3868206739425659,
+    0.2138841599225998,
+    0.39335355162620544,
+    0.21380004286766052,
+    0.429581880569458,
+    0.1465875804424286,
+    0.39559346437454224,
+    0.27372026443481445,
+    0.5427303314208984,
+    0.3526948094367981,
+    0.6388282179832458,
+    0.18703052401542664,
+    0.5684204697608948,
+    0.10398762673139572,
+    0.22804754972457886,
+    0.27390754222869873,
+    0.7250752449035645,
+    0.42387351393699646,
+    0.4317430257797241,
+    0.5105488300323486,
+    0.5738228559494019,
+    0.21982678771018982,
+    0.10559908300638199,
+    0.02478979155421257,
+    0.27520430088043213,
+    0.7023675441741943,
+    0.16033287346363068,
+    0.5399830937385559,
+    0.6034134030342102,
+    0.6642762422561646,
+    0.48320937156677246,
+    0.13037841022014618,
+    0.6289256811141968,
+    0.14109672605991364,
+    0.312267541885376,
+    0.6944162249565125,
+    0.5651903748512268,
+    0.5463401079177856,
+    0.4730447828769684,
+    0.27938708662986755,
+    0.7027943134307861,
+    0.36230146884918213,
+    0.29766759276390076,
+    0.22361616790294647,
+    0.3949165940284729,
+    0.37440937757492065,
+    0.3667985200881958,
+    0.5079476833343506,
+    0.1961221545934677,
+    0.2874886393547058,
+    0.4079972207546234,
+    0.359017014503479,
+    0.21121162176132202,
+    0.2869509160518646,
+    0.43861478567123413,
+    0.6411409974098206,
+    0.2532181143760681,
+    0.14912842214107513,
+    0.7552571892738342,
+    0.155478835105896,
+    0.5188911557197571,
+    0.1903069168329239,
+    0.3430940806865692,
+    0.3608488440513611,
+    0.4610786736011505,
+    0.15284627676010132,
+    0.43197712302207947,
+    0.09041480720043182,
+    0.579029381275177,
+    0.7106095552444458,
+    0.24629837274551392,
+    0.43898433446884155,
+    0.25902417302131653,
+    0.405537873506546,
+    0.7083703279495239,
+    0.5259550213813782,
+    0.5558066368103027,
+    0.19719471037387848,
+    0.18502306938171387,
+    0.34330320358276367,
+    0.6143893599510193,
+    0.07310193032026291,
+    0.21957609057426453,
+    0.35984084010124207,
+    0.724631667137146,
+    0.42562830448150635,
+    0.4947870969772339,
+    0.2710569500923157,
+    0.8737128973007202,
+    0.8817986845970154,
+    0.3980502188205719,
+    0.09629248082637787,
+    0.26531389355659485,
+    0.7738139033317566,
+    0.35212162137031555,
+    0.744041383266449,
+    0.35638195276260376,
+    0.14060403406620026,
+    0.20616498589515686,
+    0.7951903939247131,
+    0.34173187613487244,
+    0.48922112584114075,
+    0.628520667552948,
+    0.08161108940839767,
+    0.1048884391784668,
+    0.06372205168008804,
+    0.6272633671760559,
+    0.5529935359954834,
+    0.19295373558998108,
+    0.3965024948120117,
+    0.29236942529678345,
+    0.363751083612442,
+    0.6862728595733643,
+    0.5806248784065247,
+    0.059777386486530304,
+    0.6213939785957336,
+    0.20255210995674133,
+    0.28428879380226135,
+    0.47792965173721313,
+    0.45932480692863464,
+    0.3356798589229584,
+    0.3052414655685425,
+    0.49888938665390015,
+    0.3692249655723572,
+    0.48628538846969604,
+    0.7133625149726868,
+    0.4685496389865875,
+    0.07628084719181061,
+    0.3926492929458618,
+    0.16024909913539886,
+    0.661271333694458,
+    0.19524836540222168,
+    0.3242189884185791,
+    0.18549099564552307,
+    0.308008074760437,
+    0.04988975450396538,
+    0.35709190368652344,
+    0.4346681535243988,
+    0.10398144274950027,
+    0.9149097800254822,
+    0.9197279214859009,
+    0.044296443462371826,
+    0.5154619216918945,
+    0.8018894791603088,
+    0.1883978247642517,
+    0.2928934097290039,
+    0.23548327386379242,
+    0.3332909047603607,
+    0.0663757249712944,
+    0.4060712456703186,
+    0.18223422765731812,
+    0.5976179838180542,
+    0.627048134803772,
+    0.3943668007850647,
+    0.32953956723213196,
+    0.6619501709938049,
+    0.609294056892395,
+    0.4194333553314209,
+    0.4661709666252136,
+    0.39851081371307373,
+    0.572421133518219,
+    0.12296762317419052,
+    0.28549328446388245,
+    0.38379839062690735,
+    0.11800221353769302,
+    0.5684623718261719,
+    0.1417904645204544,
+    0.7877758741378784
+  ]
+}

p4rel-mae-base-d40-s1/exports/step_00040000/fsd50k_lora_1ep_s1.json

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-base-d40-s1/exports/step_00040000",
+  "fingerprint": "bbcd70148f3424d1",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-base-d40-s1",
+  "target": "fsd50k",
+  "num_classes": 200,
+  "lora_layers_wrapped": 48,
+  "params": {
+    "trainable": 2551496,
+    "total": 87767240,
+    "fraction": 0.02907116596124021
+  },
+  "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.001,
+    "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.40204498572275044,
+  "lora/map": 0.40204498572275044,
+  "fsd50k/map": 0.40204498572275044,
+  "train_seconds": 88.33,
+  "total_seconds": 116.65,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.024176643292885273,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.022882542308464933,
+      "val_map": 0.40204498572275044,
+      "lr": 0.0,
+      "train_s": 88.3309578258777
+    }
+  ],
+  "per_class_ap": [
+    0.2868277132511139,
+    0.09199962764978409,
+    0.5064923763275146,
+    0.3778490126132965,
+    0.4864477515220642,
+    0.6351116299629211,
+    0.8509681224822998,
+    0.5539026856422424,
+    0.5945214033126831,
+    0.4795023500919342,
+    0.3232788145542145,
+    0.40537941455841064,
+    0.23966334760189056,
+    0.49986153841018677,
+    0.5143641233444214,
+    0.4708757996559143,
+    0.023691292852163315,
+    0.18942417204380035,
+    0.3662792444229126,
+    0.20939476788043976,
+    0.5423720479011536,
+    0.3990152180194855,
+    0.7824221253395081,
+    0.11086484789848328,
+    0.21798090636730194,
+    0.43058329820632935,
+    0.33289995789527893,
+    0.2560206949710846,
+    0.7850073575973511,
+    0.3227558434009552,
+    0.6281664967536926,
+    0.5941691398620605,
+    0.4841618239879608,
+    0.1684798002243042,
+    0.35358577966690063,
+    0.23820850253105164,
+    0.42796140909194946,
+    0.21035172045230865,
+    0.4243353009223938,
+    0.23564743995666504,
+    0.558647096157074,
+    0.2677789032459259,
+    0.67221599817276,
+    0.21648618578910828,
+    0.5792478322982788,
+    0.06825024634599686,
+    0.2611943781375885,
+    0.2767676115036011,
+    0.7423682808876038,
+    0.41004693508148193,
+    0.417806476354599,
+    0.529211699962616,
+    0.5929099917411804,
+    0.15076695382595062,
+    0.1853940188884735,
+    0.022984225302934647,
+    0.30313342809677124,
+    0.6749137043952942,
+    0.11950220167636871,
+    0.5344170331954956,
+    0.598850667476654,
+    0.6680132746696472,
+    0.4874689280986786,
+    0.26255810260772705,
+    0.5841707587242126,
+    0.14060205221176147,
+    0.2665032148361206,
+    0.6868894696235657,
+    0.4788881242275238,
+    0.5309520959854126,
+    0.4726368486881256,
+    0.2649778723716736,
+    0.716117262840271,
+    0.31161177158355713,
+    0.29179278016090393,
+    0.19032219052314758,
+    0.4461700916290283,
+    0.36395496129989624,
+    0.31378328800201416,
+    0.422208696603775,
+    0.19740517437458038,
+    0.3825416564941406,
+    0.35443559288978577,
+    0.33215805888175964,
+    0.20709069073200226,
+    0.27550825476646423,
+    0.4205017685890198,
+    0.6268089413642883,
+    0.3040677607059479,
+    0.1330297291278839,
+    0.7574865818023682,
+    0.15606966614723206,
+    0.5651310682296753,
+    0.21983788907527924,
+    0.35545220971107483,
+    0.3476526439189911,
+    0.4760396182537079,
+    0.15756714344024658,
+    0.4204103648662567,
+    0.09744763374328613,
+    0.5772943496704102,
+    0.708596408367157,
+    0.25756174325942993,
+    0.4034435451030731,
+    0.2613902688026428,
+    0.36674949526786804,
+    0.6974204182624817,
+    0.529420018196106,
+    0.5635408163070679,
+    0.2453659027814865,
+    0.18613332509994507,
+    0.35762089490890503,
+    0.6562285423278809,
+    0.11117037385702133,
+    0.19955076277256012,
+    0.3760051727294922,
+    0.7513865828514099,
+    0.4530717432498932,
+    0.49069011211395264,
+    0.26777228713035583,
+    0.8727611899375916,
+    0.8812680244445801,
+    0.4115712642669678,
+    0.10970006883144379,
+    0.3004406690597534,
+    0.7686704397201538,
+    0.25476399064064026,
+    0.7446767091751099,
+    0.3503870666027069,
+    0.13977757096290588,
+    0.22351446747779846,
+    0.7952112555503845,
+    0.4338694214820862,
+    0.48409971594810486,
+    0.6424005031585693,
+    0.06646569073200226,
+    0.13309621810913086,
+    0.05746248736977577,
+    0.6775367259979248,
+    0.5417240858078003,
+    0.19760458171367645,
+    0.3691152334213257,
+    0.33738797903060913,
+    0.42925283312797546,
+    0.6622111201286316,
+    0.5865805149078369,
+    0.08122323453426361,
+    0.5987508296966553,
+    0.18731220066547394,
+    0.28117045760154724,
+    0.462756872177124,
+    0.5063198804855347,
+    0.32987552881240845,
+    0.4177740812301636,
+    0.5435790419578552,
+    0.36770039796829224,
+    0.5109778642654419,
+    0.7432087063789368,
+    0.4823584258556366,
+    0.06151190027594566,
+    0.4365471303462982,
+    0.1962539106607437,
+    0.6491353511810303,
+    0.1848086714744568,
+    0.29931676387786865,
+    0.29872894287109375,
+    0.2946300506591797,
+    0.06953775137662888,
+    0.43140000104904175,
+    0.447731077671051,
+    0.09624635428190231,
+    0.8949383497238159,
+    0.9014146327972412,
+    0.047485433518886566,
+    0.6120149493217468,
+    0.7935183048248291,
+    0.20667561888694763,
+    0.27745190262794495,
+    0.22187592089176178,
+    0.3243238031864166,
+    0.0589175708591938,
+    0.5228252410888672,
+    0.17667430639266968,
+    0.6189729571342468,
+    0.6359952092170715,
+    0.3482457399368286,
+    0.3144325315952301,
+    0.6633426547050476,
+    0.59928959608078,
+    0.4271712303161621,
+    0.44028326869010925,
+    0.44944146275520325,
+    0.5689435005187988,
+    0.14883102476596832,
+    0.37086063623428345,
+    0.30226510763168335,
+    0.10995947569608688,
+    0.6196359992027283,
+    0.13641361892223358,
+    0.7859960794448853
+  ]
+}

p4rel-mae-base-d40-s1/exports/step_00040000/fsd50k_lora_1ep_s2.json

@@ -0,0 +1,284 @@
+{
+  "export": "/workspace/runs/p4rel-mae-base-d40-s1/exports/step_00040000",
+  "fingerprint": "bbcd70148f3424d1",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-base-d40-s1",
+  "target": "fsd50k",
+  "num_classes": 200,
+  "lora_layers_wrapped": 48,
+  "params": {
+    "trainable": 2551496,
+    "total": 87767240,
+    "fraction": 0.02907116596124021
+  },
+  "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.001,
+    "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.40156575655564664,
+  "lora/map": 0.40156575655564664,
+  "fsd50k/map": 0.40156575655564664,
+  "train_seconds": 88.21,
+  "total_seconds": 116.43,
+  "history": [
+    {
+      "step": 0,
+      "epoch": 0.0,
+      "loss": null,
+      "val_map": 0.023574699426535516,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 989,
+      "epoch": 1.0,
+      "loss": 0.022883491282011472,
+      "val_map": 0.40156575655564664,
+      "lr": 0.0,
+      "train_s": 88.20865563501138
+    }
+  ],
+  "per_class_ap": [
+    0.267708420753479,
+    0.1576356291770935,
+    0.5527531504631042,
+    0.37573203444480896,
+    0.4873256981372833,
+    0.6394843459129333,
+    0.8631688356399536,
+    0.5783767104148865,
+    0.5949477553367615,
+    0.49124473333358765,
+    0.35180699825286865,
+    0.4140194058418274,
+    0.3367994725704193,
+    0.44154924154281616,
+    0.5161475539207458,
+    0.4830186069011688,
+    0.028388382866978645,
+    0.1981654316186905,
+    0.40749648213386536,
+    0.19747063517570496,
+    0.5856243371963501,
+    0.39933934807777405,
+    0.7679383754730225,
+    0.10470350086688995,
+    0.24699702858924866,
+    0.4233461618423462,
+    0.35708439350128174,
+    0.2848641276359558,
+    0.7833649516105652,
+    0.3170367479324341,
+    0.6532643437385559,
+    0.6435497999191284,
+    0.42785269021987915,
+    0.20355083048343658,
+    0.3102828562259674,
+    0.20106033980846405,
+    0.418904185295105,
+    0.2040168195962906,
+    0.46433863043785095,
+    0.2797527313232422,
+    0.5577995181083679,
+    0.3301005959510803,
+    0.6623749136924744,
+    0.18451616168022156,
+    0.5693790912628174,
+    0.13780255615711212,
+    0.2616410255432129,
+    0.27166062593460083,
+    0.7338473796844482,
+    0.4067673087120056,
+    0.368731826543808,
+    0.5247559547424316,
+    0.5688502788543701,
+    0.20597733557224274,
+    0.18782629072666168,
+    0.016344984993338585,
+    0.32730555534362793,
+    0.6813629865646362,
+    0.13413898646831512,
+    0.5658692121505737,
+    0.6137055158615112,
+    0.6549590826034546,
+    0.45457562804222107,
+    0.20637543499469757,
+    0.5192074775695801,
+    0.12745407223701477,
+    0.2544195055961609,
+    0.6596112847328186,
+    0.5138522982597351,
+    0.5850673317909241,
+    0.48131752014160156,
+    0.24388360977172852,
+    0.741733193397522,
+    0.34426912665367126,
+    0.27698761224746704,
+    0.23436178267002106,
+    0.44415709376335144,
+    0.3107660114765167,
+    0.2791781723499298,
+    0.518449604511261,
+    0.2539237439632416,
+    0.30846455693244934,
+    0.438942015171051,
+    0.2643718719482422,
+    0.21631629765033722,
+    0.27386122941970825,
+    0.4595647156238556,
+    0.641038715839386,
+    0.1984044760465622,
+    0.18361394107341766,
+    0.7666196227073669,
+    0.13098977506160736,
+    0.550981879234314,
+    0.2012113779783249,
+    0.35457172989845276,
+    0.35774728655815125,
+    0.4181545376777649,
+    0.16046936810016632,
+    0.43501442670822144,
+    0.09883470088243484,
+    0.5669639706611633,
+    0.7055339813232422,
+    0.2558593153953552,
+    0.39145469665527344,
+    0.25013813376426697,
+    0.3395631015300751,
+    0.6871054768562317,
+    0.5320224165916443,
+    0.5487054586410522,
+    0.20062845945358276,
+    0.2475309669971466,
+    0.3209775388240814,
+    0.5753768086433411,
+    0.06922495365142822,
+    0.17743366956710815,
+    0.3629584014415741,
+    0.7222537398338318,
+    0.4393894374370575,
+    0.5043013691902161,
+    0.2854171693325043,
+    0.8803263306617737,
+    0.8886353373527527,
+    0.39944273233413696,
+    0.1801549196243286,
+    0.3183618187904358,
+    0.7834023833274841,
+    0.2637978494167328,
+    0.7523576617240906,
+    0.33790653944015503,
+    0.1199038103222847,
+    0.2635002136230469,
+    0.7893633246421814,
+    0.39572805166244507,
+    0.46801114082336426,
+    0.6359232664108276,
+    0.06052551418542862,
+    0.143125519156456,
+    0.059683069586753845,
+    0.684229850769043,
+    0.54958176612854,
+    0.2330472767353058,
+    0.37239283323287964,
+    0.3202167749404907,
+    0.32866135239601135,
+    0.6799793839454651,
+    0.5476714372634888,
+    0.04794313758611679,
+    0.693553626537323,
+    0.17942893505096436,
+    0.3386240303516388,
+    0.4904811978340149,
+    0.4936044216156006,
+    0.30138325691223145,
+    0.48382458090782166,
+    0.48825156688690186,
+    0.37293675541877747,
+    0.5222029089927673,
+    0.6674729585647583,
+    0.4766770303249359,
+    0.050661832094192505,
+    0.42332690954208374,
+    0.1943310797214508,
+    0.6420796513557434,
+    0.18487684428691864,
+    0.27797719836235046,
+    0.28282755613327026,
+    0.3416304290294647,
+    0.056760720908641815,
+    0.3883839249610901,
+    0.4729307293891907,
+    0.10870547592639923,
+    0.8875117301940918,
+    0.9001509547233582,
+    0.05095858871936798,
+    0.5834493637084961,
+    0.7645660042762756,
+    0.19896480441093445,
+    0.276323527097702,
+    0.2588862478733063,
+    0.31370118260383606,
+    0.05850991979241371,
+    0.5162743926048279,
+    0.1309383660554886,
+    0.597341001033783,
+    0.6305012702941895,
+    0.3468094766139984,
+    0.3370486795902252,
+    0.6457107067108154,
+    0.6232544183731079,
+    0.44855406880378723,
+    0.4669659435749054,
+    0.42828643321990967,
+    0.5664300322532654,
+    0.12143208831548691,
+    0.2717496454715729,
+    0.3451922535896301,
+    0.10751830041408539,
+    0.5582642555236816,
+    0.14670966565608978,
+    0.7769100666046143
+  ]
+}

p4rel-mae-base-d40-s1/exports/step_00040000/lora_1ep_s0.json

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-base-d40-s1/exports/step_00040000",
+    "fingerprint": "bbcd70148f3424d1",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-base-d40-s1",
+    "lora_layers_wrapped": 48,
+    "params": {
+      "trainable": 2802959,
+      "total": 88018703,
+      "fraction": 0.03184503866183986
+    },
+    "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.001,
+      "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.0072709907253186235,
+      "val_map": 0.005655524156991202,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02434,
+      "train_map": 0.023646163915865392,
+      "val_map": null,
+      "lr": 0.0005952380952380953,
+      "train_s": 2.5
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.02365,
+      "train_map": 0.03976285231628401,
+      "val_map": null,
+      "lr": 0.0009989511174055585,
+      "train_s": 4.7
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02138,
+      "train_map": 0.06643038297087958,
+      "val_map": null,
+      "lr": 0.0009822533676115074,
+      "train_s": 6.9
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02106,
+      "train_map": 0.09146123276926238,
+      "val_map": null,
+      "lr": 0.0009458651830686461,
+      "train_s": 9.1
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02119,
+      "train_map": 0.09468714949238402,
+      "val_map": 0.05945344762770597,
+      "lr": 0.0009323333624238302,
+      "train_s": 9.8
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02071,
+      "train_map": 0.10587245693644137,
+      "val_map": null,
+      "lr": 0.0008912878610374935,
+      "train_s": 11.3
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02067,
+      "train_map": 0.12174210090639045,
+      "val_map": null,
+      "lr": 0.0008207731429852658,
+      "train_s": 13.6
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.02036,
+      "train_map": 0.1425008185391634,
+      "val_map": null,
+      "lr": 0.0007372303126360084,
+      "train_s": 15.8
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.02009,
+      "train_map": 0.15719925526412915,
+      "val_map": null,
+      "lr": 0.0006441061652589842,
+      "train_s": 18.0
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.02021,
+      "train_map": 0.16668583489989816,
+      "val_map": 0.10020599852137593,
+      "lr": 0.000589191946809388,
+      "train_s": 19.2
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01963,
+      "train_map": 0.1739850845921297,
+      "val_map": null,
+      "lr": 0.0005452428004011125,
+      "train_s": 20.2
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.01915,
+      "train_map": 0.19397340427582246,
+      "val_map": null,
+      "lr": 0.00044471910522567154,
+      "train_s": 22.4
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01919,
+      "train_map": 0.20179864777821122,
+      "val_map": null,
+      "lr": 0.0003466824685095434,
+      "train_s": 24.6
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01891,
+      "train_map": 0.22073571889818536,
+      "val_map": null,
+      "lr": 0.00025517766841304316,
+      "train_s": 26.8
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.01802,
+      "train_map": 0.22307350809210505,
+      "val_map": 0.1324912202526838,
+      "lr": 0.00018613483544815295,
+      "train_s": 28.7
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01772,
+      "train_map": 0.2266748598379134,
+      "val_map": null,
+      "lr": 0.00017397999374035568,
+      "train_s": 29.0
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01824,
+      "train_map": 0.22675666032072714,
+      "val_map": null,
+      "lr": 0.00010643948374400427,
+      "train_s": 31.2
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01828,
+      "train_map": 0.23416978371571714,
+      "val_map": null,
+      "lr": 5.53427127998607e-05,
+      "train_s": 33.4
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01815,
+      "train_map": 0.2372011621102719,
+      "val_map": null,
+      "lr": 2.2797822416041917e-05,
+      "train_s": 35.7
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.01864,
+      "train_map": 0.2383813457165896,
+      "val_map": null,
+      "lr": 1.0147543904674559e-05,
+      "train_s": 37.9
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.01845,
+      "train_map": 0.2387196580442888,
+      "val_map": 0.14243609737604856,
+      "lr": 1e-05,
+      "train_s": 38.1
+    }
+  ],
+  "final_train_map": 0.2387196580442888,
+  "final_val_map": 0.14243609737604856,
+  "lora/map": 0.14243609737604856,
+  "train_seconds": 38.1,
+  "wall_seconds": 207.6
+}

p4rel-mae-base-d40-s1/exports/step_00040000/lora_1ep_s1.json

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-base-d40-s1/exports/step_00040000",
+    "fingerprint": "bbcd70148f3424d1",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-base-d40-s1",
+    "lora_layers_wrapped": 48,
+    "params": {
+      "trainable": 2802959,
+      "total": 88018703,
+      "fraction": 0.03184503866183986
+    },
+    "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.001,
+      "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.00766083885254197,
+      "val_map": 0.005782036447413336,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.0243,
+      "train_map": 0.021514212010177364,
+      "val_map": null,
+      "lr": 0.0005952380952380953,
+      "train_s": 2.2
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.0234,
+      "train_map": 0.04562126289508827,
+      "val_map": null,
+      "lr": 0.0009989511174055585,
+      "train_s": 4.4
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02181,
+      "train_map": 0.07384026528643603,
+      "val_map": null,
+      "lr": 0.0009822533676115074,
+      "train_s": 6.6
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02189,
+      "train_map": 0.08807852343864654,
+      "val_map": null,
+      "lr": 0.0009458651830686461,
+      "train_s": 8.8
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02158,
+      "train_map": 0.09164594957141994,
+      "val_map": 0.05722888086505404,
+      "lr": 0.0009323333624238302,
+      "train_s": 9.5
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02072,
+      "train_map": 0.10743103009530218,
+      "val_map": null,
+      "lr": 0.0008912878610374935,
+      "train_s": 11.1
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02033,
+      "train_map": 0.12401364714594688,
+      "val_map": null,
+      "lr": 0.0008207731429852658,
+      "train_s": 13.3
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.02018,
+      "train_map": 0.14837864354302296,
+      "val_map": null,
+      "lr": 0.0007372303126360084,
+      "train_s": 15.5
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.01939,
+      "train_map": 0.16477008217563976,
+      "val_map": null,
+      "lr": 0.0006441061652589842,
+      "train_s": 17.7
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.01894,
+      "train_map": 0.17123931379497503,
+      "val_map": 0.10513916344026658,
+      "lr": 0.000589191946809388,
+      "train_s": 18.9
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01942,
+      "train_map": 0.17801097638426133,
+      "val_map": null,
+      "lr": 0.0005452428004011125,
+      "train_s": 19.9
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.01935,
+      "train_map": 0.18731019835905005,
+      "val_map": null,
+      "lr": 0.00044471910522567154,
+      "train_s": 22.1
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01872,
+      "train_map": 0.20819363386000894,
+      "val_map": null,
+      "lr": 0.0003466824685095434,
+      "train_s": 24.3
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01856,
+      "train_map": 0.21536340689608222,
+      "val_map": null,
+      "lr": 0.00025517766841304316,
+      "train_s": 26.5
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.01898,
+      "train_map": 0.22583255065654526,
+      "val_map": 0.13473882521829608,
+      "lr": 0.00018613483544815295,
+      "train_s": 28.4
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.0186,
+      "train_map": 0.22586899438954225,
+      "val_map": null,
+      "lr": 0.00017397999374035568,
+      "train_s": 28.7
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01812,
+      "train_map": 0.23134417278670918,
+      "val_map": null,
+      "lr": 0.00010643948374400427,
+      "train_s": 31.0
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01788,
+      "train_map": 0.23553245616599605,
+      "val_map": null,
+      "lr": 5.53427127998607e-05,
+      "train_s": 33.2
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01771,
+      "train_map": 0.23891442132834523,
+      "val_map": null,
+      "lr": 2.2797822416041917e-05,
+      "train_s": 35.4
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.01741,
+      "train_map": 0.2396214027947391,
+      "val_map": null,
+      "lr": 1.0147543904674559e-05,
+      "train_s": 37.6
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.01736,
+      "train_map": 0.23995060081358652,
+      "val_map": 0.14423925583507066,
+      "lr": 1e-05,
+      "train_s": 37.9
+    }
+  ],
+  "final_train_map": 0.23995060081358652,
+  "final_val_map": 0.14423925583507066,
+  "lora/map": 0.14423925583507066,
+  "train_seconds": 37.9,
+  "wall_seconds": 203.8
+}

p4rel-mae-base-d40-s1/exports/step_00040000/lora_1ep_s2.json

@@ -0,0 +1,260 @@
+{
+  "meta": {
+    "export": "/workspace/runs/p4rel-mae-base-d40-s1/exports/step_00040000",
+    "fingerprint": "bbcd70148f3424d1",
+    "campaign_id": "phase4-reliability",
+    "trial_id": "p4rel-mae-base-d40-s1",
+    "lora_layers_wrapped": 48,
+    "params": {
+      "trainable": 2802959,
+      "total": 88018703,
+      "fraction": 0.03184503866183986
+    },
+    "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.001,
+      "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.007086286275224333,
+      "val_map": 0.006405316863829302,
+      "lr": 0.0,
+      "train_s": 0.0
+    },
+    {
+      "step": 25,
+      "epoch": 0.0584,
+      "loss": 0.02426,
+      "train_map": 0.02299389633636363,
+      "val_map": null,
+      "lr": 0.0005952380952380953,
+      "train_s": 2.2
+    },
+    {
+      "step": 50,
+      "epoch": 0.1168,
+      "loss": 0.0244,
+      "train_map": 0.04261888122734785,
+      "val_map": null,
+      "lr": 0.0009989511174055585,
+      "train_s": 4.4
+    },
+    {
+      "step": 75,
+      "epoch": 0.1752,
+      "loss": 0.02221,
+      "train_map": 0.07351422617481439,
+      "val_map": null,
+      "lr": 0.0009822533676115074,
+      "train_s": 6.6
+    },
+    {
+      "step": 100,
+      "epoch": 0.2336,
+      "loss": 0.02143,
+      "train_map": 0.0937325205928708,
+      "val_map": null,
+      "lr": 0.0009458651830686461,
+      "train_s": 8.9
+    },
+    {
+      "step": 107,
+      "epoch": 0.25,
+      "loss": 0.02149,
+      "train_map": 0.09706945679825599,
+      "val_map": 0.0634519215412544,
+      "lr": 0.0009323333624238302,
+      "train_s": 9.5
+    },
+    {
+      "step": 125,
+      "epoch": 0.2921,
+      "loss": 0.02153,
+      "train_map": 0.10709637922872799,
+      "val_map": null,
+      "lr": 0.0008912878610374935,
+      "train_s": 11.1
+    },
+    {
+      "step": 150,
+      "epoch": 0.3505,
+      "loss": 0.02087,
+      "train_map": 0.12916821561509875,
+      "val_map": null,
+      "lr": 0.0008207731429852658,
+      "train_s": 13.3
+    },
+    {
+      "step": 175,
+      "epoch": 0.4089,
+      "loss": 0.01989,
+      "train_map": 0.14497288241646689,
+      "val_map": null,
+      "lr": 0.0007372303126360084,
+      "train_s": 15.5
+    },
+    {
+      "step": 200,
+      "epoch": 0.4673,
+      "loss": 0.01977,
+      "train_map": 0.16895011108389843,
+      "val_map": null,
+      "lr": 0.0006441061652589842,
+      "train_s": 17.7
+    },
+    {
+      "step": 214,
+      "epoch": 0.5,
+      "loss": 0.01979,
+      "train_map": 0.17304326811773557,
+      "val_map": 0.10560158892769712,
+      "lr": 0.000589191946809388,
+      "train_s": 18.9
+    },
+    {
+      "step": 225,
+      "epoch": 0.5257,
+      "loss": 0.01911,
+      "train_map": 0.17905931340519912,
+      "val_map": null,
+      "lr": 0.0005452428004011125,
+      "train_s": 19.9
+    },
+    {
+      "step": 250,
+      "epoch": 0.5841,
+      "loss": 0.01887,
+      "train_map": 0.19128282357668203,
+      "val_map": null,
+      "lr": 0.00044471910522567154,
+      "train_s": 22.1
+    },
+    {
+      "step": 275,
+      "epoch": 0.6425,
+      "loss": 0.01836,
+      "train_map": 0.20937572744722696,
+      "val_map": null,
+      "lr": 0.0003466824685095434,
+      "train_s": 24.4
+    },
+    {
+      "step": 300,
+      "epoch": 0.7009,
+      "loss": 0.01795,
+      "train_map": 0.2149496165314509,
+      "val_map": null,
+      "lr": 0.00025517766841304316,
+      "train_s": 26.6
+    },
+    {
+      "step": 321,
+      "epoch": 0.75,
+      "loss": 0.01844,
+      "train_map": 0.2235100169629,
+      "val_map": 0.13483946585048703,
+      "lr": 0.00018613483544815295,
+      "train_s": 28.4
+    },
+    {
+      "step": 325,
+      "epoch": 0.7593,
+      "loss": 0.01857,
+      "train_map": 0.2253406779165512,
+      "val_map": null,
+      "lr": 0.00017397999374035568,
+      "train_s": 28.8
+    },
+    {
+      "step": 350,
+      "epoch": 0.8178,
+      "loss": 0.01756,
+      "train_map": 0.22750273067150725,
+      "val_map": null,
+      "lr": 0.00010643948374400427,
+      "train_s": 31.0
+    },
+    {
+      "step": 375,
+      "epoch": 0.8762,
+      "loss": 0.01786,
+      "train_map": 0.23417321584793602,
+      "val_map": null,
+      "lr": 5.53427127998607e-05,
+      "train_s": 33.2
+    },
+    {
+      "step": 400,
+      "epoch": 0.9346,
+      "loss": 0.01806,
+      "train_map": 0.2377188914920251,
+      "val_map": null,
+      "lr": 2.2797822416041917e-05,
+      "train_s": 35.4
+    },
+    {
+      "step": 425,
+      "epoch": 0.993,
+      "loss": 0.01786,
+      "train_map": 0.23843197191427454,
+      "val_map": null,
+      "lr": 1.0147543904674559e-05,
+      "train_s": 37.6
+    },
+    {
+      "step": 428,
+      "epoch": 1.0,
+      "loss": 0.0181,
+      "train_map": 0.23836781411118932,
+      "val_map": 0.14540478897650339,
+      "lr": 1e-05,
+      "train_s": 37.9
+    }
+  ],
+  "final_train_map": 0.23836781411118932,
+  "final_val_map": 0.14540478897650339,
+  "lora/map": 0.14540478897650339,
+  "train_seconds": 37.9,
+  "wall_seconds": 206.3
+}

p4rel-mae-base-d40-s1/exports/step_00040000/probe.json

@@ -0,0 +1,576 @@
+{
+  "export_dir": "/workspace/runs/p4rel-mae-base-d40-s1/exports/step_00040000",
+  "fingerprint": "bbcd70148f3424d1",
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-base-d40-s1",
+  "probe/map": 0.1555549956653161,
+  "probe/map_train": 0.6510825290060134,
+  "probe/alpha": 10.0,
+  "probe/holdout_map": 0.18355205611359376,
+  "probe/pooling": "concat",
+  "probe/feature_dim": 1536,
+  "probe/fit_clips": 20550,
+  "probe/score_clips": 18886,
+  "probe/classes_scored": 527,
+  "probe/extract_s": 22.698232227994595,
+  "probe/total_s": 25.912856095004827,
+  "alpha_selection": [
+    {
+      "alpha": 0.01,
+      "holdout_map": 0.17822245074846063
+    },
+    {
+      "alpha": 0.1,
+      "holdout_map": 0.17939844790075604
+    },
+    {
+      "alpha": 1.0,
+      "holdout_map": 0.18234973372688315
+    },
+    {
+      "alpha": 10.0,
+      "holdout_map": 0.18355205611359376
+    },
+    {
+      "alpha": 100.0,
+      "holdout_map": 0.17722770876023192
+    },
+    {
+      "alpha": 1000.0,
+      "holdout_map": 0.15724063052599666
+    },
+    {
+      "alpha": 10000.0,
+      "holdout_map": 0.12435337163568023
+    }
+  ],
+  "per_class_ap": [
+    0.649660587310791,
+    0.011531878262758255,
+    0.06870786100625992,
+    0.19059664011001587,
+    0.11245746165513992,
+    0.2973753809928894,
+    0.194083571434021,
+    0.26396888494491577,
+    0.031132198870182037,
+    0.03988432511687279,
+    0.026173589751124382,
+    0.07299155741930008,
+    0.5463979244232178,
+    0.30108657479286194,
+    0.041696369647979736,
+    0.50522381067276,
+    0.3299018442630768,
+    0.3841819167137146,
+    0.06891686469316483,
+    0.11504977941513062,
+    0.2533133924007416,
+    0.10696710646152496,
+    0.20760242640972137,
+    0.1712116003036499,
+    0.09230062365531921,
+    0.04199836030602455,
+    0.08113671839237213,
+    0.18203593790531158,
+    0.26009196043014526,
+    0.11011450737714767,
+    0.26354458928108215,
+    0.16068805754184723,
+    0.016781769692897797,
+    0.06231308728456497,
+    0.280534952878952,
+    0.055261772125959396,
+    0.395977258682251,
+    0.06089138984680176,
+    0.05632761865854263,
+    0.1535310298204422,
+    0.17783164978027344,
+    0.021277327090501785,
+    0.07960410416126251,
+    0.39781656861305237,
+    0.012635128572583199,
+    0.09970595687627792,
+    0.020123813301324844,
+    0.18754340708255768,
+    0.08384900540113449,
+    0.10009579360485077,
+    0.13628679513931274,
+    0.00956714153289795,
+    0.072127565741539,
+    0.04314146563410759,
+    0.2565401494503021,
+    0.13264723122119904,
+    0.24629071354866028,
+    0.3603380620479584,
+    0.136276975274086,
+    0.0986391231417656,
+    0.16804654896259308,
+    0.20774951577186584,
+    0.05675717070698738,
+    0.13339488208293915,
+    0.5498473048210144,
+    0.24693790078163147,
+    0.28177228569984436,
+    0.34757906198501587,
+    0.39761799573898315,
+    0.36274436116218567,
+    0.10585009306669235,
+    0.1458890438079834,
+    0.22148391604423523,
+    0.20786574482917786,
+    0.20464061200618744,
+    0.25804269313812256,
+    0.06925196200609207,
+    0.2483530193567276,
+    0.19654777646064758,
+    0.1479509472846985,
+    0.03782770410180092,
+    0.14359137415885925,
+    0.3983965218067169,
+    0.14200232923030853,
+    0.1275745928287506,
+    0.0408252589404583,
+    0.1187477856874466,
+    0.0612422339618206,
+    0.05658969655632973,
+    0.1438302844762802,
+    0.15559321641921997,
+    0.15714839100837708,
+    0.2272961586713791,
+    0.0332479365170002,
+    0.05644187331199646,
+    0.12656687200069427,
+    0.030711598694324493,
+    0.05518651753664017,
+    0.1540411412715912,
+    0.24601320922374725,
+    0.1854722648859024,
+    0.31478115916252136,
+    0.06112559512257576,
+    0.059707384556531906,
+    0.021833008155226707,
+    0.0449586845934391,
+    0.07766420394182205,
+    0.06756105273962021,
+    0.09841395169496536,
+    0.19981533288955688,
+    0.24374446272850037,
+    0.14744603633880615,
+    0.13567282259464264,
+    0.10503401607275009,
+    0.04717234894633293,
+    0.2691011428833008,
+    0.2606171667575836,
+    0.12719903886318207,
+    0.15543602406978607,
+    0.35790473222732544,
+    0.3713338077068329,
+    0.24373596906661987,
+    0.11740121990442276,
+    0.08986800909042358,
+    0.013561929576098919,
+    0.04357125610113144,
+    0.13494282960891724,
+    0.16744492948055267,
+    0.18525411188602448,
+    0.1379377543926239,
+    0.0059397961013019085,
+    0.09693991392850876,
+    0.19158779084682465,
+    0.19763056933879852,
+    0.10382945835590363,
+    0.007656183559447527,
+    0.10611964762210846,
+    0.725721001625061,
+    0.24570249021053314,
+    0.31448689103126526,
+    0.4264761507511139,
+    0.3376084268093109,
+    0.14877033233642578,
+    0.1118195503950119,
+    0.10323134064674377,
+    0.1148471012711525,
+    0.09184981882572174,
+    0.30260080099105835,
+    0.14082065224647522,
+    0.19323383271694183,
+    0.22459082305431366,
+    0.16331961750984192,
+    0.24543502926826477,
+    0.11415418982505798,
+    0.1390945464372635,
+    0.2234419882297516,
+    0.1079612448811531,
+    0.08068989962339401,
+    0.01853451132774353,
+    0.06004941090941429,
+    0.4518069922924042,
+    0.33509546518325806,
+    0.40456464886665344,
+    0.14114132523536682,
+    0.30761590600013733,
+    0.3524976670742035,
+    0.3820563852787018,
+    0.20926326513290405,
+    0.3756721317768097,
+    0.3460104465484619,
+    0.23224546015262604,
+    0.35060280561447144,
+    0.3803838789463043,
+    0.2056935876607895,
+    0.1127634048461914,
+    0.08990362286567688,
+    0.12050548940896988,
+    0.3400477170944214,
+    0.07525460422039032,
+    0.14900222420692444,
+    0.18675318360328674,
+    0.16128413379192352,
+    0.146251380443573,
+    0.2577729821205139,
+    0.08365271240472794,
+    0.41460973024368286,
+    0.3815775215625763,
+    0.20246228575706482,
+    0.2388652265071869,
+    0.17188797891139984,
+    0.17584329843521118,
+    0.11434746533632278,
+    0.38733455538749695,
+    0.21310098469257355,
+    0.21377050876617432,
+    0.3875793218612671,
+    0.06796938180923462,
+    0.1271955817937851,
+    0.14327560365200043,
+    0.2827748954296112,
+    0.05128035694360733,
+    0.20219312608242035,
+    0.08344221860170364,
+    0.1973709762096405,
+    0.26454707980155945,
+    0.26824015378952026,
+    0.3301827609539032,
+    0.5154873132705688,
+    0.2826227843761444,
+    0.3302050828933716,
+    0.7520438432693481,
+    0.43208473920822144,
+    0.47872674465179443,
+    0.06383223086595535,
+    0.39875537157058716,
+    0.09260578453540802,
+    0.030561894178390503,
+    0.21042440831661224,
+    0.20439212024211884,
+    0.10850193351507187,
+    0.16924285888671875,
+    0.1015925407409668,
+    0.13817501068115234,
+    0.17861370742321014,
+    0.06598930060863495,
+    0.05454954132437706,
+    0.046835437417030334,
+    0.023564830422401428,
+    0.11954978853464127,
+    0.046279605478048325,
+    0.04667651280760765,
+    0.06316681206226349,
+    0.0850386694073677,
+    0.038751307874917984,
+    0.03383849188685417,
+    0.014254823327064514,
+    0.193876713514328,
+    0.0214709360152483,
+    0.31418514251708984,
+    0.023431694135069847,
+    0.2021256387233734,
+    0.3037566542625427,
+    0.2895669639110565,
+    0.21302427351474762,
+    0.13939706981182098,
+    0.2368183135986328,
+    0.08210774511098862,
+    0.20562846958637238,
+    0.024105392396450043,
+    0.15002790093421936,
+    0.1297411024570465,
+    0.02257746458053589,
+    0.058737341314554214,
+    0.1038106307387352,
+    0.046386685222387314,
+    0.2306145429611206,
+    0.23891474306583405,
+    0.22462394833564758,
+    0.04924370348453522,
+    0.06393241137266159,
+    0.031043700873851776,
+    0.06965477019548416,
+    0.22134126722812653,
+    0.07287193834781647,
+    0.01853613182902336,
+    0.03551645949482918,
+    0.05590555816888809,
+    0.06391431391239166,
+    0.05838346853852272,
+    0.02905394323170185,
+    0.030623076483607292,
+    0.24901339411735535,
+    0.08569569140672684,
+    0.027323246002197266,
+    0.16430264711380005,
+    0.02901170775294304,
+    0.032308511435985565,
+    0.015334689058363438,
+    0.06406573951244354,
+    0.028646187856793404,
+    0.025865845382213593,
+    0.2760612964630127,
+    0.1562701314687729,
+    0.26148027181625366,
+    0.08322788029909134,
+    0.27823570370674133,
+    0.37277311086654663,
+    0.40237173438072205,
+    0.2763063311576843,
+    0.23397991061210632,
+    0.22448360919952393,
+    0.2303609997034073,
+    0.12260687351226807,
+    0.1956704705953598,
+    0.2533000409603119,
+    0.26699498295783997,
+    0.15177035331726074,
+    0.006398755591362715,
+    0.02400691621005535,
+    0.01911940611898899,
+    0.29133349657058716,
+    0.2289622575044632,
+    0.18275675177574158,
+    0.11040524393320084,
+    0.024326477199792862,
+    0.14672674238681793,
+    0.034271009266376495,
+    0.1882290542125702,
+    0.09568117558956146,
+    0.16163095831871033,
+    0.1664183884859085,
+    0.06763830035924911,
+    0.1265750378370285,
+    0.09337668120861053,
+    0.09863689541816711,
+    0.12291721999645233,
+    0.07356400787830353,
+    0.05861014500260353,
+    0.22005979716777802,
+    0.08077646046876907,
+    0.242270365357399,
+    0.05937211960554123,
+    0.46358242630958557,
+    0.3853006958961487,
+    0.26354992389678955,
+    0.2683316469192505,
+    0.05525144934654236,
+    0.07677306979894638,
+    0.30955269932746887,
+    0.2891177535057068,
+    0.15808869898319244,
+    0.2844487130641937,
+    0.3084021210670471,
+    0.3310549855232239,
+    0.1883286088705063,
+    0.2236301153898239,
+    0.1677321046590805,
+    0.2241082340478897,
+    0.04961244389414787,
+    0.038057513535022736,
+    0.1634928435087204,
+    0.023296507075428963,
+    0.05155530571937561,
+    0.18047527968883514,
+    0.0977441594004631,
+    0.07632524520158768,
+    0.11030001193284988,
+    0.38242340087890625,
+    0.03876304626464844,
+    0.027398617938160896,
+    0.13834695518016815,
+    0.03055829554796219,
+    0.09446188062429428,
+    0.08937561511993408,
+    0.03250175714492798,
+    0.08449078351259232,
+    0.024067414924502373,
+    0.01592734083533287,
+    0.040962353348731995,
+    0.14681294560432434,
+    0.011870254762470722,
+    0.10354217886924744,
+    0.16920702159404755,
+    0.14626210927963257,
+    0.03798092156648636,
+    0.05363687500357628,
+    0.38055160641670227,
+    0.2125851958990097,
+    0.033810053020715714,
+    0.15356691181659698,
+    0.1472739428281784,
+    0.136264830827713,
+    0.1422460526227951,
+    0.3524866998195648,
+    0.18332262337207794,
+    0.06014739349484444,
+    0.08848471939563751,
+    0.12563948333263397,
+    0.03420524299144745,
+    0.019631726667284966,
+    0.038902733474969864,
+    0.10453058779239655,
+    0.23211495578289032,
+    0.10602200776338577,
+    0.18567009270191193,
+    0.204959899187088,
+    0.19788043200969696,
+    0.07846469432115555,
+    0.07141613960266113,
+    0.07569237053394318,
+    0.3271247446537018,
+    0.11829831451177597,
+    0.03241308405995369,
+    0.3967396318912506,
+    0.2723517119884491,
+    0.13383841514587402,
+    0.5161826610565186,
+    0.6004098057746887,
+    0.036753445863723755,
+    0.4409528076648712,
+    0.44104626774787903,
+    0.049376312643289566,
+    0.13294488191604614,
+    0.2319055050611496,
+    0.02562122792005539,
+    0.031996164470911026,
+    0.036989450454711914,
+    0.08537891507148743,
+    0.10703065246343613,
+    0.018768835812807083,
+    0.02196366712450981,
+    0.11046794056892395,
+    0.03164341673254967,
+    0.04865419119596481,
+    0.3209114670753479,
+    0.2107369303703308,
+    0.0341898612678051,
+    0.06410712748765945,
+    0.11164602637290955,
+    0.13188929855823517,
+    0.2599509656429291,
+    0.06694723665714264,
+    0.09519106894731522,
+    0.22300730645656586,
+    0.11061511188745499,
+    0.11436401307582855,
+    0.049839459359645844,
+    0.2314661294221878,
+    0.27565789222717285,
+    0.18247398734092712,
+    0.16282084584236145,
+    0.18346573412418365,
+    0.2653260827064514,
+    0.17294453084468842,
+    0.007649125065654516,
+    0.10569779574871063,
+    0.2172165960073471,
+    0.09046309441328049,
+    0.11111524701118469,
+    0.06055905297398567,
+    0.02903774008154869,
+    0.07971115410327911,
+    0.0158844031393528,
+    0.11479806900024414,
+    0.01622716337442398,
+    0.13137073814868927,
+    0.043420806527137756,
+    0.007104760501533747,
+    0.024455014616250992,
+    0.10824909061193466,
+    0.013613712042570114,
+    0.07048037648200989,
+    0.019546350464224815,
+    0.05693713575601578,
+    0.07354298233985901,
+    0.24852274358272552,
+    0.04586284980177879,
+    0.14327070116996765,
+    0.06975352764129639,
+    0.15353922545909882,
+    0.0296296626329422,
+    0.043900907039642334,
+    0.41100412607192993,
+    0.43581241369247437,
+    0.23199258744716644,
+    0.1498725265264511,
+    0.02981048822402954,
+    0.15409177541732788,
+    0.05898803845047951,
+    0.1377081423997879,
+    0.16032972931861877,
+    0.05709007754921913,
+    0.13860651850700378,
+    0.015274768695235252,
+    0.10008939355611801,
+    0.006756240501999855,
+    0.28882163763046265,
+    0.08473503589630127,
+    0.0822983980178833,
+    0.25617262721061707,
+    0.18517599999904633,
+    0.13403746485710144,
+    0.02735348604619503,
+    0.1455976665019989,
+    0.08278953284025192,
+    0.014882870018482208,
+    0.00720117287710309,
+    0.06928342580795288,
+    0.047795508056879044,
+    0.04776627570390701,
+    0.21504850685596466,
+    0.3386775851249695,
+    0.11093848198652267,
+    0.15141136944293976,
+    0.23415839672088623,
+    0.03365979343652725,
+    0.03660109266638756,
+    0.06338979303836823,
+    0.05852808058261871,
+    0.12436946481466293,
+    0.11487988382577896,
+    0.33208802342414856,
+    0.006971671245992184,
+    0.13503284752368927,
+    0.021383903920650482,
+    0.010312278755009174,
+    0.09864304214715958,
+    0.017986437305808067,
+    0.021182449534535408,
+    0.042374469339847565,
+    0.025857767090201378,
+    0.14909563958644867,
+    0.08024438470602036,
+    0.008163845166563988,
+    0.15762515366077423,
+    0.29603058099746704,
+    0.08618886768817902,
+    0.35150015354156494,
+    0.04010431095957756,
+    0.020978597924113274,
+    0.03046293929219246,
+    0.24243201315402985,
+    0.18371877074241638,
+    0.06187931448221207,
+    0.01555029209703207,
+    0.14447060227394104,
+    0.11610310524702072
+  ]
+}

p4rel-mae-base-d40-s1/mixture_plan.npy

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

p4rel-mae-base-d40-s1/run.json

@@ -0,0 +1,94 @@
+{
+  "campaign_id": "phase4-reliability",
+  "trial_id": "p4rel-mae-base-d40-s1",
+  "fingerprint": "bbcd70148f3424d1",
+  "wandb_run_id": "phase4-reliability-p4rel-mae-base-d40-s1",
+  "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": 85252608,
+    "decoder_params": 2585856,
+    "trainable_params": 88035328,
+    "total_params": 88035328
+  },
+  "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": 2.0000019000511308e-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": 1786976009.7243702,
+  "finished_at": 1786977600.4665067,
+  "summary": {
+    "completed": true,
+    "final_step": 40000,
+    "clips_seen": 1920000,
+    "peak_gib": 2.0947370529174805,
+    "val/loss": 0.6301599419675767,
+    "val/frame": 0.6301599419675767,
+    "val/utterance": 0.0
+  }
+}

p4rel-mae-base-d40-s1/status.json

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