quinnlue's picture
Upload folder using huggingface_hub
77d8715 verified
Raw
History Blame
2.3 kB
{
"campaign_id": "phase5",
"trial_id": "p5-mae-15m-d182",
"fingerprint": "5d7463349b08a290",
"wandb_run_id": "phase5-p5-mae-15m-d182",
"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.499994756876356,
"distinct_clips": 738331,
"capped_clusters": 0,
"realized_weights": [
0.0004411458333333333,
0.0205703125,
0.3725114583333333,
6.25e-05,
0.14009166666666667,
0.0,
0.019627083333333333,
0.00044166666666666665,
0.013650520833333334,
0.012203125,
0.029566666666666668,
0.13380104166666668,
0.002328125,
0.011072395833333333,
0.0,
0.0,
0.033088020833333336,
0.13021458333333333,
0.042083854166666664,
0.03824583333333333
]
},
"started_at": 1787009590.7095032,
"finished_at": 1787010317.2786963,
"summary": {
"completed": true,
"final_step": 40000,
"clips_seen": 1920000,
"peak_gib": 0.6756577491760254,
"val/loss": 0.6331442929804325,
"val/frame": 0.6331442929804325,
"val/utterance": 0.0
}
}