quinnlue's picture
Upload folder using huggingface_hub
6bb4167 verified
Raw
History Blame
2.09 kB
{
"campaign_id": "phase5",
"trial_id": "p5-mae-15m-d129",
"fingerprint": "24f41a90eea0382c",
"wandb_run_id": "phase5-p5-mae-15m-d129",
"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": 6.735312156141205,
"distinct_clips": 409228,
"capped_clusters": 0,
"realized_weights": [
0.0,
0.0,
0.0,
0.0,
0.19392395833333334,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.3889432291666667,
0.4171328125,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0
]
},
"started_at": 1787008297.9943821,
"finished_at": 1787009002.0978668,
"summary": {
"completed": true,
"final_step": 40000,
"clips_seen": 1920000,
"peak_gib": 0.6756577491760254,
"val/loss": 0.6347254221327603,
"val/frame": 0.6347254221327603,
"val/utterance": 0.0
}
}