quinnlue's picture
Upload folder using huggingface_hub
7f13f78 verified
Raw
History Blame
2.19 kB
{
"campaign_id": "phase5",
"trial_id": "p5-mae-1m-d112",
"fingerprint": "836c6f5cfddf0584",
"wandb_run_id": "phase5-p5-mae-1m-d112",
"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": 1039968,
"decoder_params": 91728,
"trainable_params": 1168816,
"total_params": 1168816
},
"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": 4.0000038001022615e-06,
"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": 555179,
"capped_clusters": 0,
"realized_weights": [
0.008534895833333334,
0.007994270833333334,
0.0,
0.24986302083333334,
0.09022083333333333,
0.00078125,
0.15417395833333333,
0.0,
0.0,
0.3723046875,
0.0,
0.0,
0.0,
0.0,
0.03516197916666667,
0.0009630208333333333,
0.0001796875,
0.0,
0.07982239583333334,
0.0
]
},
"started_at": 1786987807.080491,
"finished_at": 1786988289.4497201,
"summary": {
"completed": true,
"final_step": 40000,
"clips_seen": 1920000,
"peak_gib": 0.29828882217407227,
"val/loss": 0.6430522981099784,
"val/frame": 0.6430522981099784,
"val/utterance": 0.0
}
}