quinnlue's picture
Upload folder using huggingface_hub
ec9629f verified
Raw
History Blame
15.9 kB
{
"meta": {
"export": "/workspace/runs/p5-mae-15m-d110/exports/step_00020000",
"fingerprint": "40ce4d5afc2dd106",
"campaign_id": "phase5",
"trial_id": "p5-mae-15m-d110",
"lora_layers_wrapped": 32,
"params": {
"trainable": 1002383,
"total": 15284879,
"fraction": 0.0655800415561026
},
"steps_per_epoch": 428,
"total_steps": 1712,
"train_clips": 20550,
"val_clips": 18886,
"config": {
"rank": 16,
"alpha": 32.0,
"targets": [
"qkv",
"proj",
"fc1",
"fc2"
],
"train_norms": true,
"epochs": 4.0,
"batch_size": 48,
"lr": 0.003,
"min_lr": 1e-05,
"warmup_frac": 0.1,
"betas": [
0.9,
0.95
],
"weight_decay": 0.05,
"clip_norm": 4.0,
"mixup": 0.0,
"mixup_prob": 0.9,
"freq_mask": 0,
"time_mask": 0,
"roll_aug": false,
"mask_ratio": 0.0,
"num_classes": 527,
"prior_bias_init": true,
"num_workers": 8,
"seed": 2,
"train_probe_clips": 4096,
"train_eval_every": 25,
"val_evals": 5,
"eval_batch_size": 128
},
"gpu": "NVIDIA GeForce RTX 4090"
},
"history": [
{
"step": 0,
"epoch": 0.0,
"loss": null,
"train_map": 0.007121785415294076,
"val_map": 0.006121005405463363,
"lr": 0.0,
"train_s": 0.0
},
{
"step": 25,
"epoch": 0.0584,
"loss": 0.02477,
"train_map": 0.014894589853996103,
"val_map": null,
"lr": 0.0004385964912280702,
"train_s": 0.7
},
{
"step": 50,
"epoch": 0.1168,
"loss": 0.02388,
"train_map": 0.035247838792088045,
"val_map": null,
"lr": 0.0008771929824561404,
"train_s": 1.3
},
{
"step": 75,
"epoch": 0.1752,
"loss": 0.02244,
"train_map": 0.05803766907426923,
"val_map": null,
"lr": 0.0013157894736842105,
"train_s": 1.9
},
{
"step": 100,
"epoch": 0.2336,
"loss": 0.02171,
"train_map": 0.07180234314202273,
"val_map": null,
"lr": 0.0017543859649122807,
"train_s": 2.5
},
{
"step": 125,
"epoch": 0.2921,
"loss": 0.02169,
"train_map": 0.0737651140195807,
"val_map": null,
"lr": 0.0021929824561403508,
"train_s": 3.1
},
{
"step": 150,
"epoch": 0.3505,
"loss": 0.02147,
"train_map": 0.08660082944246465,
"val_map": null,
"lr": 0.002631578947368421,
"train_s": 3.8
},
{
"step": 175,
"epoch": 0.4089,
"loss": 0.02145,
"train_map": 0.09539002549845153,
"val_map": null,
"lr": 0.002999950292348948,
"train_s": 4.3
},
{
"step": 200,
"epoch": 0.4673,
"loss": 0.02058,
"train_map": 0.10232892341335681,
"val_map": null,
"lr": 0.002997387988070006,
"train_s": 4.9
},
{
"step": 225,
"epoch": 0.5257,
"loss": 0.02037,
"train_map": 0.11319764946857767,
"val_map": null,
"lr": 0.0029909498761254527,
"train_s": 5.5
},
{
"step": 250,
"epoch": 0.5841,
"loss": 0.02033,
"train_map": 0.125714667512111,
"val_map": null,
"lr": 0.0029806526766201318,
"train_s": 6.1
},
{
"step": 275,
"epoch": 0.6425,
"loss": 0.02073,
"train_map": 0.12563644074154506,
"val_map": null,
"lr": 0.0029665231319059614,
"train_s": 6.7
},
{
"step": 300,
"epoch": 0.7009,
"loss": 0.02047,
"train_map": 0.12006251172664481,
"val_map": null,
"lr": 0.002948597937130687,
"train_s": 7.3
},
{
"step": 325,
"epoch": 0.7593,
"loss": 0.02031,
"train_map": 0.1391861233144532,
"val_map": null,
"lr": 0.0029269236449387106,
"train_s": 7.8
},
{
"step": 350,
"epoch": 0.8178,
"loss": 0.01995,
"train_map": 0.14788485125707465,
"val_map": null,
"lr": 0.002901556544571499,
"train_s": 8.4
},
{
"step": 375,
"epoch": 0.8762,
"loss": 0.02004,
"train_map": 0.15370363936683543,
"val_map": null,
"lr": 0.002872562515681558,
"train_s": 9.0
},
{
"step": 400,
"epoch": 0.9346,
"loss": 0.01963,
"train_map": 0.15289603819782568,
"val_map": null,
"lr": 0.0028400168572396164,
"train_s": 9.6
},
{
"step": 425,
"epoch": 0.993,
"loss": 0.02018,
"train_map": 0.15185503932369954,
"val_map": null,
"lr": 0.002804004091979359,
"train_s": 10.1
},
{
"step": 428,
"epoch": 1.0,
"loss": 0.02059,
"train_map": 0.15695323427985342,
"val_map": 0.0889595405267555,
"lr": 0.002799453919445244,
"train_s": 10.2
},
{
"step": 450,
"epoch": 1.0514,
"loss": 0.01872,
"train_map": 0.17179214600933118,
"val_map": null,
"lr": 0.0027646177468875835,
"train_s": 10.8
},
{
"step": 475,
"epoch": 1.1098,
"loss": 0.01832,
"train_map": 0.16923329111478475,
"val_map": null,
"lr": 0.0027219601103098453,
"train_s": 11.4
},
{
"step": 500,
"epoch": 1.1682,
"loss": 0.01876,
"train_map": 0.17817886400105434,
"val_map": null,
"lr": 0.002676141966302412,
"train_s": 12.0
},
{
"step": 525,
"epoch": 1.2266,
"loss": 0.01867,
"train_map": 0.18780806936514502,
"val_map": null,
"lr": 0.002627282306920426,
"train_s": 12.6
},
{
"step": 550,
"epoch": 1.285,
"loss": 0.01903,
"train_map": 0.18351549328529262,
"val_map": null,
"lr": 0.002575508023189468,
"train_s": 13.2
},
{
"step": 575,
"epoch": 1.3435,
"loss": 0.01871,
"train_map": 0.18800017271505445,
"val_map": null,
"lr": 0.0025209535755631,
"train_s": 13.7
},
{
"step": 600,
"epoch": 1.4019,
"loss": 0.01908,
"train_map": 0.1971410638780865,
"val_map": null,
"lr": 0.0024637606447222133,
"train_s": 14.3
},
{
"step": 625,
"epoch": 1.4603,
"loss": 0.01884,
"train_map": 0.1875621410589524,
"val_map": null,
"lr": 0.002404077763623079,
"train_s": 14.9
},
{
"step": 650,
"epoch": 1.5187,
"loss": 0.01822,
"train_map": 0.20243293067981158,
"val_map": null,
"lr": 0.002342059931749691,
"train_s": 15.4
},
{
"step": 675,
"epoch": 1.5771,
"loss": 0.01827,
"train_map": 0.21267284584572746,
"val_map": null,
"lr": 0.002277868212572206,
"train_s": 16.0
},
{
"step": 700,
"epoch": 1.6355,
"loss": 0.01905,
"train_map": 0.21491943431604632,
"val_map": null,
"lr": 0.002211669315256905,
"train_s": 16.6
},
{
"step": 725,
"epoch": 1.6939,
"loss": 0.01864,
"train_map": 0.2236736006160007,
"val_map": null,
"lr": 0.0021436351617139878,
"train_s": 17.2
},
{
"step": 750,
"epoch": 1.7523,
"loss": 0.01823,
"train_map": 0.22503249822855986,
"val_map": null,
"lr": 0.0020739424401076136,
"train_s": 17.8
},
{
"step": 775,
"epoch": 1.8107,
"loss": 0.0182,
"train_map": 0.23521185626030194,
"val_map": null,
"lr": 0.0020027721459877326,
"train_s": 18.4
},
{
"step": 800,
"epoch": 1.8692,
"loss": 0.01807,
"train_map": 0.23918549506969228,
"val_map": null,
"lr": 0.0019303091122354168,
"train_s": 18.9
},
{
"step": 825,
"epoch": 1.9276,
"loss": 0.01823,
"train_map": 0.24216639422113517,
"val_map": null,
"lr": 0.0018567415290424508,
"train_s": 19.5
},
{
"step": 850,
"epoch": 1.986,
"loss": 0.01859,
"train_map": 0.24795692841534767,
"val_map": null,
"lr": 0.0017822604551718095,
"train_s": 20.1
},
{
"step": 856,
"epoch": 2.0,
"loss": 0.01784,
"train_map": 0.24720295727168135,
"val_map": 0.13462025389177779,
"lr": 0.0017642705177736725,
"train_s": 20.2
},
{
"step": 875,
"epoch": 2.0444,
"loss": 0.01742,
"train_map": 0.2535018431176511,
"val_map": null,
"lr": 0.0017070593217683146,
"train_s": 20.7
},
{
"step": 900,
"epoch": 2.1028,
"loss": 0.01709,
"train_map": 0.26097485659894953,
"val_map": null,
"lr": 0.0016313334300080934,
"train_s": 21.3
},
{
"step": 925,
"epoch": 2.1612,
"loss": 0.01617,
"train_map": 0.27669691193346957,
"val_map": null,
"lr": 0.0015552794438914697,
"train_s": 21.9
},
{
"step": 950,
"epoch": 2.2196,
"loss": 0.01633,
"train_map": 0.2710613943795467,
"val_map": null,
"lr": 0.0014790948794965334,
"train_s": 22.6
},
{
"step": 975,
"epoch": 2.278,
"loss": 0.01622,
"train_map": 0.2787023775139912,
"val_map": null,
"lr": 0.0014029775920198172,
"train_s": 23.2
},
{
"step": 1000,
"epoch": 2.3364,
"loss": 0.01631,
"train_map": 0.2969034678802827,
"val_map": null,
"lr": 0.0013271252619362684,
"train_s": 23.8
},
{
"step": 1025,
"epoch": 2.3949,
"loss": 0.01672,
"train_map": 0.3068938201622051,
"val_map": null,
"lr": 0.0012517348816129803,
"train_s": 24.4
},
{
"step": 1050,
"epoch": 2.4533,
"loss": 0.01632,
"train_map": 0.31181753200586654,
"val_map": null,
"lr": 0.0011770022437099775,
"train_s": 25.0
},
{
"step": 1075,
"epoch": 2.5117,
"loss": 0.01606,
"train_map": 0.32037331831007065,
"val_map": null,
"lr": 0.0011031214326966937,
"train_s": 25.6
},
{
"step": 1100,
"epoch": 2.5701,
"loss": 0.01606,
"train_map": 0.33173145944927684,
"val_map": null,
"lr": 0.0010302843208047179,
"train_s": 26.2
},
{
"step": 1125,
"epoch": 2.6285,
"loss": 0.01587,
"train_map": 0.33801908223612315,
"val_map": null,
"lr": 0.000958680069725827,
"train_s": 26.9
},
{
"step": 1150,
"epoch": 2.6869,
"loss": 0.01594,
"train_map": 0.34766395610935663,
"val_map": null,
"lr": 0.0008884946393494339,
"train_s": 27.5
},
{
"step": 1175,
"epoch": 2.7453,
"loss": 0.01627,
"train_map": 0.3529564615946963,
"val_map": null,
"lr": 0.0008199103048152753,
"train_s": 28.1
},
{
"step": 1200,
"epoch": 2.8037,
"loss": 0.01573,
"train_map": 0.371245008343707,
"val_map": null,
"lr": 0.0007531051831355856,
"train_s": 28.7
},
{
"step": 1225,
"epoch": 2.8621,
"loss": 0.01578,
"train_map": 0.36930742310943254,
"val_map": null,
"lr": 0.0006882527706161351,
"train_s": 29.3
},
{
"step": 1250,
"epoch": 2.9206,
"loss": 0.01548,
"train_map": 0.37726545430705716,
"val_map": null,
"lr": 0.0006255214922774859,
"train_s": 29.9
},
{
"step": 1275,
"epoch": 2.979,
"loss": 0.01514,
"train_map": 0.3875832997891607,
"val_map": null,
"lr": 0.0005650742644466275,
"train_s": 30.4
},
{
"step": 1284,
"epoch": 3.0,
"loss": 0.01527,
"train_map": 0.38977456795832804,
"val_map": 0.1848591136387364,
"lr": 0.0005439028207490256,
"train_s": 30.6
},
{
"step": 1300,
"epoch": 3.0374,
"loss": 0.01399,
"train_map": 0.3982927684214468,
"val_map": null,
"lr": 0.0005070680716549702,
"train_s": 31.0
},
{
"step": 1325,
"epoch": 3.0958,
"loss": 0.01361,
"train_map": 0.41116537747151827,
"val_map": null,
"lr": 0.0004516535589415209,
"train_s": 31.6
},
{
"step": 1350,
"epoch": 3.1542,
"loss": 0.01329,
"train_map": 0.41983100466882595,
"val_map": null,
"lr": 0.0003989746406200353,
"train_s": 32.2
},
{
"step": 1375,
"epoch": 3.2126,
"loss": 0.0138,
"train_map": 0.42213304830358084,
"val_map": null,
"lr": 0.0003491681265262127,
"train_s": 32.7
},
{
"step": 1400,
"epoch": 3.271,
"loss": 0.01352,
"train_map": 0.42976896196914804,
"val_map": null,
"lr": 0.00030236336671557766,
"train_s": 33.3
},
{
"step": 1425,
"epoch": 3.3294,
"loss": 0.01308,
"train_map": 0.4382105492182656,
"val_map": null,
"lr": 0.00025868191553479617,
"train_s": 33.9
},
{
"step": 1450,
"epoch": 3.3879,
"loss": 0.01309,
"train_map": 0.4441180522474098,
"val_map": null,
"lr": 0.00021823721593883938,
"train_s": 34.5
},
{
"step": 1475,
"epoch": 3.4463,
"loss": 0.01288,
"train_map": 0.4463726418266034,
"val_map": null,
"lr": 0.00018113430487385143,
"train_s": 35.1
},
{
"step": 1500,
"epoch": 3.5047,
"loss": 0.01332,
"train_map": 0.44990806522261373,
"val_map": null,
"lr": 0.00014746954049084314,
"train_s": 35.7
},
{
"step": 1525,
"epoch": 3.5631,
"loss": 0.01359,
"train_map": 0.45210007852759826,
"val_map": null,
"lr": 0.0001173303518986653,
"train_s": 36.3
},
{
"step": 1550,
"epoch": 3.6215,
"loss": 0.01322,
"train_map": 0.4580144303156314,
"val_map": null,
"lr": 9.079501210615714e-05,
"train_s": 36.9
},
{
"step": 1575,
"epoch": 3.6799,
"loss": 0.01314,
"train_map": 0.46015735469984714,
"val_map": null,
"lr": 6.793243474315096e-05,
"train_s": 37.5
},
{
"step": 1600,
"epoch": 3.7383,
"loss": 0.01316,
"train_map": 0.4605217744477899,
"val_map": null,
"lr": 4.8801995088266645e-05,
"train_s": 38.1
},
{
"step": 1625,
"epoch": 3.7967,
"loss": 0.01298,
"train_map": 0.4617658964087886,
"val_map": null,
"lr": 3.345337586828326e-05,
"train_s": 38.7
},
{
"step": 1650,
"epoch": 3.8551,
"loss": 0.01304,
"train_map": 0.46350407681694744,
"val_map": null,
"lr": 2.1926438229568824e-05,
"train_s": 39.3
},
{
"step": 1675,
"epoch": 3.9136,
"loss": 0.01265,
"train_map": 0.4637404012986842,
"val_map": null,
"lr": 1.4251118216649036e-05,
"train_s": 39.9
},
{
"step": 1700,
"epoch": 3.972,
"loss": 0.01306,
"train_map": 0.4639172537969796,
"val_map": null,
"lr": 1.0447349026775707e-05,
"train_s": 40.5
},
{
"step": 1712,
"epoch": 4.0,
"loss": 0.01318,
"train_map": 0.46437664578476145,
"val_map": 0.19552661800682827,
"lr": 1e-05,
"train_s": 40.8
}
],
"final_train_map": 0.46437664578476145,
"final_val_map": 0.19552661800682827,
"lora/map": 0.19552661800682827,
"train_seconds": 40.8,
"wall_seconds": 153.9
}