quinnlue's picture
Upload folder using huggingface_hub
161866f verified
Raw
History Blame
2.28 kB
{
"campaign_id": "phase5",
"trial_id": "p5-mae-base-d0",
"fingerprint": "1e529a361ad8274e",
"wandb_run_id": "phase5-p5-mae-base-d0",
"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": 5.443241878168011,
"distinct_clips": 793378,
"capped_clusters": 0,
"realized_weights": [
0.16161145833333335,
0.042833854166666664,
0.066421875,
0.0,
0.0354609375,
0.0,
0.027091666666666667,
0.0008380208333333333,
0.02515625,
0.0256796875,
0.0,
0.0005234375,
0.31433020833333336,
0.01475,
0.19293958333333333,
0.00020885416666666668,
0.00020885416666666668,
0.0655859375,
0.005177604166666666,
0.021181770833333332
]
},
"started_at": 1787019781.1505105,
"finished_at": 1787021421.9695146,
"summary": {
"completed": true,
"final_step": 40000,
"clips_seen": 1920000,
"peak_gib": 2.078686237335205,
"val/loss": 0.6272649462334812,
"val/frame": 0.6272649462334812,
"val/utterance": 0.0
}
}