quinnlue's picture
Upload folder using huggingface_hub
efe13b5 verified
Raw
History Blame
2.15 kB
{
"campaign_id": "phase5",
"trial_id": "p5-mae-base-d10",
"fingerprint": "5a27946c4fe591a3",
"wandb_run_id": "phase5-p5-mae-base-d10",
"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.500010975743606,
"distinct_clips": 395130,
"capped_clusters": 0,
"realized_weights": [
0.1977421875,
0.20976354166666666,
0.0,
0.0,
0.0,
0.03904114583333333,
0.0,
0.17794947916666667,
0.0,
0.3723046875,
0.0014333333333333333,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.001765625,
0.0
]
},
"started_at": 1787020387.4760597,
"finished_at": 1787022050.3035717,
"summary": {
"completed": true,
"final_step": 40000,
"clips_seen": 1920000,
"peak_gib": 2.078686237335205,
"val/loss": 0.6308394414372742,
"val/frame": 0.6308394414372742,
"val/utterance": 0.0
}
}