quinnlue's picture
Upload folder using huggingface_hub
ec9629f verified
Raw
History Blame
15.8 kB
{
"meta": {
"export": "/workspace/runs/p5-mae-15m-d110/exports/step_00040000",
"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.00747152418339982,
"val_map": 0.006283473859424482,
"lr": 0.0,
"train_s": 0.0
},
{
"step": 25,
"epoch": 0.0584,
"loss": 0.02469,
"train_map": 0.016734078041948486,
"val_map": null,
"lr": 0.0004385964912280702,
"train_s": 0.8
},
{
"step": 50,
"epoch": 0.1168,
"loss": 0.02376,
"train_map": 0.038401560629700335,
"val_map": null,
"lr": 0.0008771929824561404,
"train_s": 1.4
},
{
"step": 75,
"epoch": 0.1752,
"loss": 0.02234,
"train_map": 0.062274035181456106,
"val_map": null,
"lr": 0.0013157894736842105,
"train_s": 2.1
},
{
"step": 100,
"epoch": 0.2336,
"loss": 0.02156,
"train_map": 0.0763700886451033,
"val_map": null,
"lr": 0.0017543859649122807,
"train_s": 2.7
},
{
"step": 125,
"epoch": 0.2921,
"loss": 0.02147,
"train_map": 0.0786219381495773,
"val_map": null,
"lr": 0.0021929824561403508,
"train_s": 3.3
},
{
"step": 150,
"epoch": 0.3505,
"loss": 0.02136,
"train_map": 0.09451149635598333,
"val_map": null,
"lr": 0.002631578947368421,
"train_s": 3.9
},
{
"step": 175,
"epoch": 0.4089,
"loss": 0.02127,
"train_map": 0.09829963836642212,
"val_map": null,
"lr": 0.002999950292348948,
"train_s": 4.5
},
{
"step": 200,
"epoch": 0.4673,
"loss": 0.02041,
"train_map": 0.11040406151978649,
"val_map": null,
"lr": 0.002997387988070006,
"train_s": 5.1
},
{
"step": 225,
"epoch": 0.5257,
"loss": 0.02027,
"train_map": 0.10278193046828375,
"val_map": null,
"lr": 0.0029909498761254527,
"train_s": 5.6
},
{
"step": 250,
"epoch": 0.5841,
"loss": 0.02026,
"train_map": 0.11937080134383175,
"val_map": null,
"lr": 0.0029806526766201318,
"train_s": 6.2
},
{
"step": 275,
"epoch": 0.6425,
"loss": 0.02054,
"train_map": 0.12684719525436663,
"val_map": null,
"lr": 0.0029665231319059614,
"train_s": 6.8
},
{
"step": 300,
"epoch": 0.7009,
"loss": 0.02051,
"train_map": 0.13495325585659346,
"val_map": null,
"lr": 0.002948597937130687,
"train_s": 7.4
},
{
"step": 325,
"epoch": 0.7593,
"loss": 0.01995,
"train_map": 0.13484888435500467,
"val_map": null,
"lr": 0.0029269236449387106,
"train_s": 7.9
},
{
"step": 350,
"epoch": 0.8178,
"loss": 0.01988,
"train_map": 0.1519408785336893,
"val_map": null,
"lr": 0.002901556544571499,
"train_s": 8.5
},
{
"step": 375,
"epoch": 0.8762,
"loss": 0.02,
"train_map": 0.158024035217912,
"val_map": null,
"lr": 0.002872562515681558,
"train_s": 9.1
},
{
"step": 400,
"epoch": 0.9346,
"loss": 0.0196,
"train_map": 0.15509707212317347,
"val_map": null,
"lr": 0.0028400168572396164,
"train_s": 9.7
},
{
"step": 425,
"epoch": 0.993,
"loss": 0.01998,
"train_map": 0.14885244252331611,
"val_map": null,
"lr": 0.002804004091979359,
"train_s": 10.3
},
{
"step": 428,
"epoch": 1.0,
"loss": 0.02043,
"train_map": 0.1700098120039276,
"val_map": 0.09936046333934058,
"lr": 0.002799453919445244,
"train_s": 10.4
},
{
"step": 450,
"epoch": 1.0514,
"loss": 0.01867,
"train_map": 0.178933458165946,
"val_map": null,
"lr": 0.0027646177468875835,
"train_s": 10.9
},
{
"step": 475,
"epoch": 1.1098,
"loss": 0.01826,
"train_map": 0.1886663803030194,
"val_map": null,
"lr": 0.0027219601103098453,
"train_s": 11.5
},
{
"step": 500,
"epoch": 1.1682,
"loss": 0.01862,
"train_map": 0.19315558005445962,
"val_map": null,
"lr": 0.002676141966302412,
"train_s": 12.1
},
{
"step": 525,
"epoch": 1.2266,
"loss": 0.01852,
"train_map": 0.18559089782638952,
"val_map": null,
"lr": 0.002627282306920426,
"train_s": 12.7
},
{
"step": 550,
"epoch": 1.285,
"loss": 0.0187,
"train_map": 0.18455649218769424,
"val_map": null,
"lr": 0.002575508023189468,
"train_s": 13.3
},
{
"step": 575,
"epoch": 1.3435,
"loss": 0.01855,
"train_map": 0.19686756531051422,
"val_map": null,
"lr": 0.0025209535755631,
"train_s": 13.9
},
{
"step": 600,
"epoch": 1.4019,
"loss": 0.01893,
"train_map": 0.20148922516381196,
"val_map": null,
"lr": 0.0024637606447222133,
"train_s": 14.5
},
{
"step": 625,
"epoch": 1.4603,
"loss": 0.01879,
"train_map": 0.20336631112564107,
"val_map": null,
"lr": 0.002404077763623079,
"train_s": 15.1
},
{
"step": 650,
"epoch": 1.5187,
"loss": 0.01801,
"train_map": 0.20822185039806554,
"val_map": null,
"lr": 0.002342059931749691,
"train_s": 15.7
},
{
"step": 675,
"epoch": 1.5771,
"loss": 0.01825,
"train_map": 0.22070250147335105,
"val_map": null,
"lr": 0.002277868212572206,
"train_s": 16.3
},
{
"step": 700,
"epoch": 1.6355,
"loss": 0.01893,
"train_map": 0.22539490452071606,
"val_map": null,
"lr": 0.002211669315256905,
"train_s": 16.9
},
{
"step": 725,
"epoch": 1.6939,
"loss": 0.0184,
"train_map": 0.2347671391642997,
"val_map": null,
"lr": 0.0021436351617139878,
"train_s": 17.5
},
{
"step": 750,
"epoch": 1.7523,
"loss": 0.01807,
"train_map": 0.22804392986017255,
"val_map": null,
"lr": 0.0020739424401076136,
"train_s": 18.1
},
{
"step": 775,
"epoch": 1.8107,
"loss": 0.01804,
"train_map": 0.2411054527180259,
"val_map": null,
"lr": 0.0020027721459877326,
"train_s": 18.7
},
{
"step": 800,
"epoch": 1.8692,
"loss": 0.01784,
"train_map": 0.24603704484515387,
"val_map": null,
"lr": 0.0019303091122354168,
"train_s": 19.3
},
{
"step": 825,
"epoch": 1.9276,
"loss": 0.01816,
"train_map": 0.25157101162969076,
"val_map": null,
"lr": 0.0018567415290424508,
"train_s": 19.9
},
{
"step": 850,
"epoch": 1.986,
"loss": 0.0184,
"train_map": 0.2560333275117786,
"val_map": null,
"lr": 0.0017822604551718095,
"train_s": 20.4
},
{
"step": 856,
"epoch": 2.0,
"loss": 0.01773,
"train_map": 0.25158443821477483,
"val_map": 0.14386098434533606,
"lr": 0.0017642705177736725,
"train_s": 20.6
},
{
"step": 875,
"epoch": 2.0444,
"loss": 0.01712,
"train_map": 0.2695424880415371,
"val_map": null,
"lr": 0.0017070593217683146,
"train_s": 21.1
},
{
"step": 900,
"epoch": 2.1028,
"loss": 0.01668,
"train_map": 0.2717403030408232,
"val_map": null,
"lr": 0.0016313334300080934,
"train_s": 21.7
},
{
"step": 925,
"epoch": 2.1612,
"loss": 0.01583,
"train_map": 0.28776639239929713,
"val_map": null,
"lr": 0.0015552794438914697,
"train_s": 22.3
},
{
"step": 950,
"epoch": 2.2196,
"loss": 0.01638,
"train_map": 0.29574293947815755,
"val_map": null,
"lr": 0.0014790948794965334,
"train_s": 23.0
},
{
"step": 975,
"epoch": 2.278,
"loss": 0.01601,
"train_map": 0.29363738544165124,
"val_map": null,
"lr": 0.0014029775920198172,
"train_s": 23.6
},
{
"step": 1000,
"epoch": 2.3364,
"loss": 0.01604,
"train_map": 0.3071555611191484,
"val_map": null,
"lr": 0.0013271252619362684,
"train_s": 24.2
},
{
"step": 1025,
"epoch": 2.3949,
"loss": 0.01646,
"train_map": 0.3234735856882422,
"val_map": null,
"lr": 0.0012517348816129803,
"train_s": 24.9
},
{
"step": 1050,
"epoch": 2.4533,
"loss": 0.01607,
"train_map": 0.33057873270896737,
"val_map": null,
"lr": 0.0011770022437099775,
"train_s": 25.6
},
{
"step": 1075,
"epoch": 2.5117,
"loss": 0.01571,
"train_map": 0.3334367552196052,
"val_map": null,
"lr": 0.0011031214326966937,
"train_s": 26.2
},
{
"step": 1100,
"epoch": 2.5701,
"loss": 0.01579,
"train_map": 0.34554950382553423,
"val_map": null,
"lr": 0.0010302843208047179,
"train_s": 26.8
},
{
"step": 1125,
"epoch": 2.6285,
"loss": 0.0156,
"train_map": 0.34772083815363936,
"val_map": null,
"lr": 0.000958680069725827,
"train_s": 27.4
},
{
"step": 1150,
"epoch": 2.6869,
"loss": 0.01564,
"train_map": 0.3563169801264888,
"val_map": null,
"lr": 0.0008884946393494339,
"train_s": 28.1
},
{
"step": 1175,
"epoch": 2.7453,
"loss": 0.01593,
"train_map": 0.36847595263986477,
"val_map": null,
"lr": 0.0008199103048152753,
"train_s": 28.7
},
{
"step": 1200,
"epoch": 2.8037,
"loss": 0.01557,
"train_map": 0.37961104000429047,
"val_map": null,
"lr": 0.0007531051831355856,
"train_s": 29.4
},
{
"step": 1225,
"epoch": 2.8621,
"loss": 0.01565,
"train_map": 0.38286033511670764,
"val_map": null,
"lr": 0.0006882527706161351,
"train_s": 30.0
},
{
"step": 1250,
"epoch": 2.9206,
"loss": 0.01523,
"train_map": 0.3950595792022444,
"val_map": null,
"lr": 0.0006255214922774859,
"train_s": 30.6
},
{
"step": 1275,
"epoch": 2.979,
"loss": 0.01485,
"train_map": 0.4027463728933113,
"val_map": null,
"lr": 0.0005650742644466275,
"train_s": 31.2
},
{
"step": 1284,
"epoch": 3.0,
"loss": 0.01485,
"train_map": 0.4088922792045396,
"val_map": 0.19555360061934737,
"lr": 0.0005439028207490256,
"train_s": 31.4
},
{
"step": 1300,
"epoch": 3.0374,
"loss": 0.01343,
"train_map": 0.41371074226637156,
"val_map": null,
"lr": 0.0005070680716549702,
"train_s": 31.8
},
{
"step": 1325,
"epoch": 3.0958,
"loss": 0.01335,
"train_map": 0.4235385470021608,
"val_map": null,
"lr": 0.0004516535589415209,
"train_s": 32.4
},
{
"step": 1350,
"epoch": 3.1542,
"loss": 0.01311,
"train_map": 0.4305111750203982,
"val_map": null,
"lr": 0.0003989746406200353,
"train_s": 33.0
},
{
"step": 1375,
"epoch": 3.2126,
"loss": 0.01355,
"train_map": 0.43889549965890695,
"val_map": null,
"lr": 0.0003491681265262127,
"train_s": 33.5
},
{
"step": 1400,
"epoch": 3.271,
"loss": 0.01328,
"train_map": 0.4435999966500437,
"val_map": null,
"lr": 0.00030236336671557766,
"train_s": 34.1
},
{
"step": 1425,
"epoch": 3.3294,
"loss": 0.01297,
"train_map": 0.447359591178186,
"val_map": null,
"lr": 0.00025868191553479617,
"train_s": 34.7
},
{
"step": 1450,
"epoch": 3.3879,
"loss": 0.01273,
"train_map": 0.454836309224763,
"val_map": null,
"lr": 0.00021823721593883938,
"train_s": 35.3
},
{
"step": 1475,
"epoch": 3.4463,
"loss": 0.0125,
"train_map": 0.4555130274771965,
"val_map": null,
"lr": 0.00018113430487385143,
"train_s": 35.9
},
{
"step": 1500,
"epoch": 3.5047,
"loss": 0.01294,
"train_map": 0.46135252946722666,
"val_map": null,
"lr": 0.00014746954049084314,
"train_s": 36.5
},
{
"step": 1525,
"epoch": 3.5631,
"loss": 0.01349,
"train_map": 0.4642205413685408,
"val_map": null,
"lr": 0.0001173303518986653,
"train_s": 37.1
},
{
"step": 1550,
"epoch": 3.6215,
"loss": 0.01297,
"train_map": 0.46691327726563875,
"val_map": null,
"lr": 9.079501210615714e-05,
"train_s": 37.7
},
{
"step": 1575,
"epoch": 3.6799,
"loss": 0.01274,
"train_map": 0.47150502038635383,
"val_map": null,
"lr": 6.793243474315096e-05,
"train_s": 38.2
},
{
"step": 1600,
"epoch": 3.7383,
"loss": 0.01289,
"train_map": 0.4730103305793691,
"val_map": null,
"lr": 4.8801995088266645e-05,
"train_s": 38.8
},
{
"step": 1625,
"epoch": 3.7967,
"loss": 0.0128,
"train_map": 0.474199068062785,
"val_map": null,
"lr": 3.345337586828326e-05,
"train_s": 39.4
},
{
"step": 1650,
"epoch": 3.8551,
"loss": 0.01272,
"train_map": 0.4748551175351726,
"val_map": null,
"lr": 2.1926438229568824e-05,
"train_s": 40.0
},
{
"step": 1675,
"epoch": 3.9136,
"loss": 0.01244,
"train_map": 0.4754239120282077,
"val_map": null,
"lr": 1.4251118216649036e-05,
"train_s": 40.6
},
{
"step": 1700,
"epoch": 3.972,
"loss": 0.01275,
"train_map": 0.4758721256462514,
"val_map": null,
"lr": 1.0447349026775707e-05,
"train_s": 41.1
},
{
"step": 1712,
"epoch": 4.0,
"loss": 0.01266,
"train_map": 0.4761564440431473,
"val_map": 0.20647433014017524,
"lr": 1e-05,
"train_s": 41.4
}
],
"final_train_map": 0.4761564440431473,
"final_val_map": 0.20647433014017524,
"lora/map": 0.20647433014017524,
"train_seconds": 41.4,
"wall_seconds": 155.3
}