sanchit-gandhi's picture
End of training
467af1f verified
raw
history blame
53.1 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.0,
"eval_steps": 500,
"global_step": 297,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.010101010101010102,
"grad_norm": 1.4315768480300903,
"learning_rate": 2.98989898989899e-05,
"loss": 10.0726,
"step": 1
},
{
"epoch": 0.020202020202020204,
"grad_norm": 1.5613890886306763,
"learning_rate": 2.97979797979798e-05,
"loss": 10.1289,
"step": 2
},
{
"epoch": 0.030303030303030304,
"grad_norm": 1.4086546897888184,
"learning_rate": 2.96969696969697e-05,
"loss": 10.0402,
"step": 3
},
{
"epoch": 0.04040404040404041,
"grad_norm": 1.508208155632019,
"learning_rate": 2.9595959595959595e-05,
"loss": 10.0206,
"step": 4
},
{
"epoch": 0.050505050505050504,
"grad_norm": 1.6154247522354126,
"learning_rate": 2.9494949494949495e-05,
"loss": 10.0272,
"step": 5
},
{
"epoch": 0.06060606060606061,
"grad_norm": 1.8789210319519043,
"learning_rate": 2.9393939393939394e-05,
"loss": 10.0711,
"step": 6
},
{
"epoch": 0.0707070707070707,
"grad_norm": 1.879401683807373,
"learning_rate": 2.9292929292929294e-05,
"loss": 9.9488,
"step": 7
},
{
"epoch": 0.08080808080808081,
"grad_norm": 1.9335699081420898,
"learning_rate": 2.9191919191919193e-05,
"loss": 9.8084,
"step": 8
},
{
"epoch": 0.09090909090909091,
"grad_norm": 1.9985203742980957,
"learning_rate": 2.9090909090909093e-05,
"loss": 9.7694,
"step": 9
},
{
"epoch": 0.10101010101010101,
"grad_norm": 2.0941152572631836,
"learning_rate": 2.8989898989898992e-05,
"loss": 9.7268,
"step": 10
},
{
"epoch": 0.1111111111111111,
"grad_norm": 2.344831943511963,
"learning_rate": 2.8888888888888888e-05,
"loss": 9.7139,
"step": 11
},
{
"epoch": 0.12121212121212122,
"grad_norm": 2.27956485748291,
"learning_rate": 2.8787878787878788e-05,
"loss": 9.5774,
"step": 12
},
{
"epoch": 0.13131313131313133,
"grad_norm": 2.440272331237793,
"learning_rate": 2.8686868686868687e-05,
"loss": 9.4884,
"step": 13
},
{
"epoch": 0.1414141414141414,
"grad_norm": 2.7465505599975586,
"learning_rate": 2.8585858585858587e-05,
"loss": 9.4327,
"step": 14
},
{
"epoch": 0.15151515151515152,
"grad_norm": 2.5827443599700928,
"learning_rate": 2.8484848484848486e-05,
"loss": 9.2545,
"step": 15
},
{
"epoch": 0.16161616161616163,
"grad_norm": 2.7065556049346924,
"learning_rate": 2.8383838383838386e-05,
"loss": 9.2304,
"step": 16
},
{
"epoch": 0.1717171717171717,
"grad_norm": 2.944704294204712,
"learning_rate": 2.8282828282828285e-05,
"loss": 9.1568,
"step": 17
},
{
"epoch": 0.18181818181818182,
"grad_norm": 2.667189598083496,
"learning_rate": 2.8181818181818185e-05,
"loss": 9.0014,
"step": 18
},
{
"epoch": 0.1919191919191919,
"grad_norm": 2.928550958633423,
"learning_rate": 2.808080808080808e-05,
"loss": 8.9312,
"step": 19
},
{
"epoch": 0.20202020202020202,
"grad_norm": 2.930938482284546,
"learning_rate": 2.797979797979798e-05,
"loss": 8.8656,
"step": 20
},
{
"epoch": 0.21212121212121213,
"grad_norm": 2.750476121902466,
"learning_rate": 2.787878787878788e-05,
"loss": 8.7607,
"step": 21
},
{
"epoch": 0.2222222222222222,
"grad_norm": 2.8917646408081055,
"learning_rate": 2.777777777777778e-05,
"loss": 8.699,
"step": 22
},
{
"epoch": 0.23232323232323232,
"grad_norm": 2.8639166355133057,
"learning_rate": 2.767676767676768e-05,
"loss": 8.5703,
"step": 23
},
{
"epoch": 0.24242424242424243,
"grad_norm": 2.9358019828796387,
"learning_rate": 2.7575757575757578e-05,
"loss": 8.519,
"step": 24
},
{
"epoch": 0.25252525252525254,
"grad_norm": 3.063692092895508,
"learning_rate": 2.7474747474747478e-05,
"loss": 8.3564,
"step": 25
},
{
"epoch": 0.26262626262626265,
"grad_norm": 2.702965259552002,
"learning_rate": 2.7373737373737374e-05,
"loss": 8.3403,
"step": 26
},
{
"epoch": 0.2727272727272727,
"grad_norm": 2.8974392414093018,
"learning_rate": 2.7272727272727273e-05,
"loss": 8.217,
"step": 27
},
{
"epoch": 0.2828282828282828,
"grad_norm": 3.0662784576416016,
"learning_rate": 2.7171717171717173e-05,
"loss": 8.1575,
"step": 28
},
{
"epoch": 0.29292929292929293,
"grad_norm": 2.8070247173309326,
"learning_rate": 2.7070707070707072e-05,
"loss": 8.0814,
"step": 29
},
{
"epoch": 0.30303030303030304,
"grad_norm": 2.507457971572876,
"learning_rate": 2.696969696969697e-05,
"loss": 8.0387,
"step": 30
},
{
"epoch": 0.31313131313131315,
"grad_norm": 2.7476532459259033,
"learning_rate": 2.686868686868687e-05,
"loss": 7.96,
"step": 31
},
{
"epoch": 0.32323232323232326,
"grad_norm": 2.3512086868286133,
"learning_rate": 2.676767676767677e-05,
"loss": 7.8822,
"step": 32
},
{
"epoch": 0.3333333333333333,
"grad_norm": 2.1365387439727783,
"learning_rate": 2.6666666666666667e-05,
"loss": 7.8282,
"step": 33
},
{
"epoch": 0.3434343434343434,
"grad_norm": 2.2421772480010986,
"learning_rate": 2.6565656565656566e-05,
"loss": 7.7613,
"step": 34
},
{
"epoch": 0.35353535353535354,
"grad_norm": 2.111851930618286,
"learning_rate": 2.6464646464646466e-05,
"loss": 7.7516,
"step": 35
},
{
"epoch": 0.36363636363636365,
"grad_norm": 2.2756099700927734,
"learning_rate": 2.6363636363636365e-05,
"loss": 7.6066,
"step": 36
},
{
"epoch": 0.37373737373737376,
"grad_norm": 1.5493675470352173,
"learning_rate": 2.6262626262626265e-05,
"loss": 7.6497,
"step": 37
},
{
"epoch": 0.3838383838383838,
"grad_norm": 1.7187950611114502,
"learning_rate": 2.6161616161616164e-05,
"loss": 7.5753,
"step": 38
},
{
"epoch": 0.3939393939393939,
"grad_norm": 1.5378119945526123,
"learning_rate": 2.6060606060606063e-05,
"loss": 7.591,
"step": 39
},
{
"epoch": 0.40404040404040403,
"grad_norm": 1.3718913793563843,
"learning_rate": 2.595959595959596e-05,
"loss": 7.6196,
"step": 40
},
{
"epoch": 0.41414141414141414,
"grad_norm": 1.9546277523040771,
"learning_rate": 2.585858585858586e-05,
"loss": 7.3466,
"step": 41
},
{
"epoch": 0.42424242424242425,
"grad_norm": 1.3100489377975464,
"learning_rate": 2.575757575757576e-05,
"loss": 7.588,
"step": 42
},
{
"epoch": 0.43434343434343436,
"grad_norm": 1.1814428567886353,
"learning_rate": 2.5656565656565658e-05,
"loss": 7.392,
"step": 43
},
{
"epoch": 0.4444444444444444,
"grad_norm": 1.3818390369415283,
"learning_rate": 2.5555555555555557e-05,
"loss": 7.413,
"step": 44
},
{
"epoch": 0.45454545454545453,
"grad_norm": 1.725190281867981,
"learning_rate": 2.5454545454545457e-05,
"loss": 7.1846,
"step": 45
},
{
"epoch": 0.46464646464646464,
"grad_norm": 1.098010778427124,
"learning_rate": 2.5353535353535356e-05,
"loss": 7.384,
"step": 46
},
{
"epoch": 0.47474747474747475,
"grad_norm": 1.1586623191833496,
"learning_rate": 2.5252525252525256e-05,
"loss": 7.2481,
"step": 47
},
{
"epoch": 0.48484848484848486,
"grad_norm": 1.3093281984329224,
"learning_rate": 2.5151515151515152e-05,
"loss": 7.1453,
"step": 48
},
{
"epoch": 0.494949494949495,
"grad_norm": 1.2143803834915161,
"learning_rate": 2.505050505050505e-05,
"loss": 7.1706,
"step": 49
},
{
"epoch": 0.5050505050505051,
"grad_norm": 1.4759756326675415,
"learning_rate": 2.494949494949495e-05,
"loss": 7.1918,
"step": 50
},
{
"epoch": 0.5151515151515151,
"grad_norm": 1.170322060585022,
"learning_rate": 2.484848484848485e-05,
"loss": 7.1158,
"step": 51
},
{
"epoch": 0.5252525252525253,
"grad_norm": 1.025293231010437,
"learning_rate": 2.474747474747475e-05,
"loss": 7.085,
"step": 52
},
{
"epoch": 0.5353535353535354,
"grad_norm": 1.673163890838623,
"learning_rate": 2.464646464646465e-05,
"loss": 7.4093,
"step": 53
},
{
"epoch": 0.5454545454545454,
"grad_norm": 0.892270565032959,
"learning_rate": 2.454545454545455e-05,
"loss": 7.1903,
"step": 54
},
{
"epoch": 0.5555555555555556,
"grad_norm": 1.1452521085739136,
"learning_rate": 2.4444444444444445e-05,
"loss": 7.2127,
"step": 55
},
{
"epoch": 0.5656565656565656,
"grad_norm": 1.4096879959106445,
"learning_rate": 2.4343434343434344e-05,
"loss": 7.2507,
"step": 56
},
{
"epoch": 0.5757575757575758,
"grad_norm": 1.3830610513687134,
"learning_rate": 2.4242424242424244e-05,
"loss": 7.2779,
"step": 57
},
{
"epoch": 0.5858585858585859,
"grad_norm": 0.88773512840271,
"learning_rate": 2.4141414141414143e-05,
"loss": 7.1313,
"step": 58
},
{
"epoch": 0.5959595959595959,
"grad_norm": 1.5023548603057861,
"learning_rate": 2.4040404040404043e-05,
"loss": 7.3029,
"step": 59
},
{
"epoch": 0.6060606060606061,
"grad_norm": 0.9497267603874207,
"learning_rate": 2.3939393939393942e-05,
"loss": 7.1915,
"step": 60
},
{
"epoch": 0.6161616161616161,
"grad_norm": 1.035706639289856,
"learning_rate": 2.3838383838383842e-05,
"loss": 7.2144,
"step": 61
},
{
"epoch": 0.6262626262626263,
"grad_norm": 0.7967200875282288,
"learning_rate": 2.3737373737373738e-05,
"loss": 7.1566,
"step": 62
},
{
"epoch": 0.6363636363636364,
"grad_norm": 0.7237263917922974,
"learning_rate": 2.3636363636363637e-05,
"loss": 7.1931,
"step": 63
},
{
"epoch": 0.6464646464646465,
"grad_norm": 1.0147182941436768,
"learning_rate": 2.3535353535353537e-05,
"loss": 7.2676,
"step": 64
},
{
"epoch": 0.6565656565656566,
"grad_norm": 0.8428776264190674,
"learning_rate": 2.3434343434343436e-05,
"loss": 7.0818,
"step": 65
},
{
"epoch": 0.6666666666666666,
"grad_norm": 0.9503577947616577,
"learning_rate": 2.3333333333333336e-05,
"loss": 7.0711,
"step": 66
},
{
"epoch": 0.6767676767676768,
"grad_norm": 0.8867577314376831,
"learning_rate": 2.3232323232323235e-05,
"loss": 7.0556,
"step": 67
},
{
"epoch": 0.6868686868686869,
"grad_norm": 0.8943458795547485,
"learning_rate": 2.3131313131313135e-05,
"loss": 7.2772,
"step": 68
},
{
"epoch": 0.696969696969697,
"grad_norm": 0.8544779419898987,
"learning_rate": 2.303030303030303e-05,
"loss": 7.0846,
"step": 69
},
{
"epoch": 0.7070707070707071,
"grad_norm": 0.8547456860542297,
"learning_rate": 2.292929292929293e-05,
"loss": 7.092,
"step": 70
},
{
"epoch": 0.7171717171717171,
"grad_norm": 0.924784779548645,
"learning_rate": 2.282828282828283e-05,
"loss": 7.0622,
"step": 71
},
{
"epoch": 0.7272727272727273,
"grad_norm": 1.1316595077514648,
"learning_rate": 2.272727272727273e-05,
"loss": 7.0545,
"step": 72
},
{
"epoch": 0.7373737373737373,
"grad_norm": NaN,
"learning_rate": 2.272727272727273e-05,
"loss": 0.0,
"step": 73
},
{
"epoch": 0.7474747474747475,
"grad_norm": 0.7388734221458435,
"learning_rate": 2.262626262626263e-05,
"loss": 7.0892,
"step": 74
},
{
"epoch": 0.7575757575757576,
"grad_norm": 1.9678359031677246,
"learning_rate": 2.2525252525252525e-05,
"loss": 7.0092,
"step": 75
},
{
"epoch": 0.7676767676767676,
"grad_norm": 0.736301064491272,
"learning_rate": 2.2424242424242424e-05,
"loss": 7.1827,
"step": 76
},
{
"epoch": 0.7777777777777778,
"grad_norm": 0.7673182487487793,
"learning_rate": 2.2323232323232324e-05,
"loss": 7.0768,
"step": 77
},
{
"epoch": 0.7878787878787878,
"grad_norm": 1.4001637697219849,
"learning_rate": 2.222222222222222e-05,
"loss": 6.9754,
"step": 78
},
{
"epoch": 0.797979797979798,
"grad_norm": 0.8279094696044922,
"learning_rate": 2.212121212121212e-05,
"loss": 7.1222,
"step": 79
},
{
"epoch": 0.8080808080808081,
"grad_norm": 0.6914481520652771,
"learning_rate": 2.202020202020202e-05,
"loss": 7.0812,
"step": 80
},
{
"epoch": 0.8181818181818182,
"grad_norm": 0.7313191294670105,
"learning_rate": 2.191919191919192e-05,
"loss": 7.05,
"step": 81
},
{
"epoch": 0.8282828282828283,
"grad_norm": 1.4300901889801025,
"learning_rate": 2.1818181818181818e-05,
"loss": 7.2278,
"step": 82
},
{
"epoch": 0.8383838383838383,
"grad_norm": 1.2713080644607544,
"learning_rate": 2.1717171717171717e-05,
"loss": 6.9029,
"step": 83
},
{
"epoch": 0.8484848484848485,
"grad_norm": 0.814030110836029,
"learning_rate": 2.1616161616161617e-05,
"loss": 7.1241,
"step": 84
},
{
"epoch": 0.8585858585858586,
"grad_norm": 1.1318764686584473,
"learning_rate": 2.1515151515151513e-05,
"loss": 7.1621,
"step": 85
},
{
"epoch": 0.8686868686868687,
"grad_norm": 1.0205698013305664,
"learning_rate": 2.1414141414141412e-05,
"loss": 7.0894,
"step": 86
},
{
"epoch": 0.8787878787878788,
"grad_norm": 0.5895187854766846,
"learning_rate": 2.1313131313131312e-05,
"loss": 6.9731,
"step": 87
},
{
"epoch": 0.8888888888888888,
"grad_norm": 0.5949335098266602,
"learning_rate": 2.121212121212121e-05,
"loss": 6.9764,
"step": 88
},
{
"epoch": 0.898989898989899,
"grad_norm": 0.909782886505127,
"learning_rate": 2.111111111111111e-05,
"loss": 6.8954,
"step": 89
},
{
"epoch": 0.9090909090909091,
"grad_norm": 0.6604152321815491,
"learning_rate": 2.101010101010101e-05,
"loss": 7.015,
"step": 90
},
{
"epoch": 0.9191919191919192,
"grad_norm": 1.1579092741012573,
"learning_rate": 2.090909090909091e-05,
"loss": 7.1838,
"step": 91
},
{
"epoch": 0.9292929292929293,
"grad_norm": 1.1899914741516113,
"learning_rate": 2.0808080808080806e-05,
"loss": 7.196,
"step": 92
},
{
"epoch": 0.9393939393939394,
"grad_norm": 1.013912558555603,
"learning_rate": 2.0707070707070705e-05,
"loss": 7.1402,
"step": 93
},
{
"epoch": 0.9494949494949495,
"grad_norm": 0.907071053981781,
"learning_rate": 2.0606060606060605e-05,
"loss": 7.129,
"step": 94
},
{
"epoch": 0.9595959595959596,
"grad_norm": 0.7546154260635376,
"learning_rate": 2.0505050505050504e-05,
"loss": 7.0195,
"step": 95
},
{
"epoch": 0.9696969696969697,
"grad_norm": 0.662983775138855,
"learning_rate": 2.0404040404040404e-05,
"loss": 7.0075,
"step": 96
},
{
"epoch": 0.9797979797979798,
"grad_norm": 0.673218846321106,
"learning_rate": 2.0303030303030303e-05,
"loss": 7.0797,
"step": 97
},
{
"epoch": 0.98989898989899,
"grad_norm": 1.617126703262329,
"learning_rate": 2.0202020202020203e-05,
"loss": 6.8117,
"step": 98
},
{
"epoch": 1.0,
"grad_norm": 1.0646132230758667,
"learning_rate": 2.01010101010101e-05,
"loss": 7.174,
"step": 99
},
{
"epoch": 1.0,
"eval_audio_cosine_sim": 0.383101224899292,
"eval_loss": 4.921389102935791,
"eval_runtime": 153.7637,
"eval_samples_per_second": 0.059,
"eval_steps_per_second": 0.02,
"eval_text_cosine_sim": 0.16359001398086548,
"step": 99
},
{
"epoch": 1.0101010101010102,
"grad_norm": 0.7128923535346985,
"learning_rate": 1.9999999999999998e-05,
"loss": 7.0139,
"step": 100
},
{
"epoch": 1.02020202020202,
"grad_norm": 0.8200396299362183,
"learning_rate": 1.9898989898989898e-05,
"loss": 7.2107,
"step": 101
},
{
"epoch": 1.0303030303030303,
"grad_norm": 0.6839468479156494,
"learning_rate": 1.9797979797979797e-05,
"loss": 7.1309,
"step": 102
},
{
"epoch": 1.0404040404040404,
"grad_norm": 1.4797368049621582,
"learning_rate": 1.9696969696969697e-05,
"loss": 6.9112,
"step": 103
},
{
"epoch": 1.0505050505050506,
"grad_norm": 0.7093661427497864,
"learning_rate": 1.9595959595959596e-05,
"loss": 7.1014,
"step": 104
},
{
"epoch": 1.0606060606060606,
"grad_norm": 1.2138644456863403,
"learning_rate": 1.9494949494949496e-05,
"loss": 6.9265,
"step": 105
},
{
"epoch": 1.0707070707070707,
"grad_norm": 0.8460476994514465,
"learning_rate": 1.9393939393939395e-05,
"loss": 7.2191,
"step": 106
},
{
"epoch": 1.0808080808080809,
"grad_norm": 0.7066294550895691,
"learning_rate": 1.929292929292929e-05,
"loss": 7.1066,
"step": 107
},
{
"epoch": 1.0909090909090908,
"grad_norm": 1.7321295738220215,
"learning_rate": 1.919191919191919e-05,
"loss": 6.8371,
"step": 108
},
{
"epoch": 1.101010101010101,
"grad_norm": 0.9022343158721924,
"learning_rate": 1.909090909090909e-05,
"loss": 6.9655,
"step": 109
},
{
"epoch": 1.1111111111111112,
"grad_norm": 0.6070654988288879,
"learning_rate": 1.898989898989899e-05,
"loss": 6.9956,
"step": 110
},
{
"epoch": 1.121212121212121,
"grad_norm": 0.6426708698272705,
"learning_rate": 1.888888888888889e-05,
"loss": 6.964,
"step": 111
},
{
"epoch": 1.1313131313131313,
"grad_norm": 2.1155552864074707,
"learning_rate": 1.878787878787879e-05,
"loss": 6.7698,
"step": 112
},
{
"epoch": 1.1414141414141414,
"grad_norm": 0.9715363383293152,
"learning_rate": 1.8686868686868688e-05,
"loss": 7.1096,
"step": 113
},
{
"epoch": 1.1515151515151516,
"grad_norm": 1.2896918058395386,
"learning_rate": 1.8585858585858584e-05,
"loss": 6.7545,
"step": 114
},
{
"epoch": 1.1616161616161615,
"grad_norm": 0.6836603283882141,
"learning_rate": 1.8484848484848484e-05,
"loss": 7.042,
"step": 115
},
{
"epoch": 1.1717171717171717,
"grad_norm": 0.6922496557235718,
"learning_rate": 1.8383838383838383e-05,
"loss": 6.9593,
"step": 116
},
{
"epoch": 1.1818181818181819,
"grad_norm": 1.5327054262161255,
"learning_rate": 1.8282828282828283e-05,
"loss": 6.7356,
"step": 117
},
{
"epoch": 1.1919191919191918,
"grad_norm": 0.686898410320282,
"learning_rate": 1.8181818181818182e-05,
"loss": 6.8844,
"step": 118
},
{
"epoch": 1.202020202020202,
"grad_norm": 2.4868946075439453,
"learning_rate": 1.808080808080808e-05,
"loss": 7.2632,
"step": 119
},
{
"epoch": 1.2121212121212122,
"grad_norm": 0.705142080783844,
"learning_rate": 1.797979797979798e-05,
"loss": 6.9367,
"step": 120
},
{
"epoch": 1.2222222222222223,
"grad_norm": 0.6701662540435791,
"learning_rate": 1.7878787878787877e-05,
"loss": 6.8656,
"step": 121
},
{
"epoch": 1.2323232323232323,
"grad_norm": 2.3600728511810303,
"learning_rate": 1.7777777777777777e-05,
"loss": 7.2908,
"step": 122
},
{
"epoch": 1.2424242424242424,
"grad_norm": 0.729280948638916,
"learning_rate": 1.7676767676767676e-05,
"loss": 6.9778,
"step": 123
},
{
"epoch": 1.2525252525252526,
"grad_norm": 0.9563952684402466,
"learning_rate": 1.7575757575757576e-05,
"loss": 6.8367,
"step": 124
},
{
"epoch": 1.2626262626262625,
"grad_norm": 0.679764449596405,
"learning_rate": 1.7474747474747475e-05,
"loss": 6.9984,
"step": 125
},
{
"epoch": 1.2727272727272727,
"grad_norm": 1.1830710172653198,
"learning_rate": 1.7373737373737375e-05,
"loss": 7.0285,
"step": 126
},
{
"epoch": 1.2828282828282829,
"grad_norm": 0.7950251698493958,
"learning_rate": 1.7272727272727274e-05,
"loss": 7.0085,
"step": 127
},
{
"epoch": 1.2929292929292928,
"grad_norm": 0.8154851794242859,
"learning_rate": 1.717171717171717e-05,
"loss": 6.8398,
"step": 128
},
{
"epoch": 1.303030303030303,
"grad_norm": 1.1023356914520264,
"learning_rate": 1.707070707070707e-05,
"loss": 7.1022,
"step": 129
},
{
"epoch": 1.3131313131313131,
"grad_norm": 1.013553500175476,
"learning_rate": 1.696969696969697e-05,
"loss": 6.8513,
"step": 130
},
{
"epoch": 1.3232323232323233,
"grad_norm": 1.4370501041412354,
"learning_rate": 1.686868686868687e-05,
"loss": 7.208,
"step": 131
},
{
"epoch": 1.3333333333333333,
"grad_norm": 0.7258216738700867,
"learning_rate": 1.6767676767676768e-05,
"loss": 6.954,
"step": 132
},
{
"epoch": 1.3434343434343434,
"grad_norm": 0.7258408069610596,
"learning_rate": 1.6666666666666667e-05,
"loss": 7.0258,
"step": 133
},
{
"epoch": 1.3535353535353536,
"grad_norm": 0.8000563383102417,
"learning_rate": 1.6565656565656567e-05,
"loss": 7.0238,
"step": 134
},
{
"epoch": 1.3636363636363638,
"grad_norm": 0.9891355633735657,
"learning_rate": 1.6464646464646466e-05,
"loss": 6.9668,
"step": 135
},
{
"epoch": 1.3737373737373737,
"grad_norm": 0.7241990566253662,
"learning_rate": 1.6363636363636363e-05,
"loss": 7.0859,
"step": 136
},
{
"epoch": 1.3838383838383839,
"grad_norm": 0.5333958268165588,
"learning_rate": 1.6262626262626262e-05,
"loss": 7.0339,
"step": 137
},
{
"epoch": 1.393939393939394,
"grad_norm": 0.7480002641677856,
"learning_rate": 1.616161616161616e-05,
"loss": 6.9745,
"step": 138
},
{
"epoch": 1.404040404040404,
"grad_norm": 0.8260339498519897,
"learning_rate": 1.606060606060606e-05,
"loss": 7.0461,
"step": 139
},
{
"epoch": 1.4141414141414141,
"grad_norm": 0.7007152438163757,
"learning_rate": 1.595959595959596e-05,
"loss": 7.046,
"step": 140
},
{
"epoch": 1.4242424242424243,
"grad_norm": 1.532183051109314,
"learning_rate": 1.585858585858586e-05,
"loss": 6.8063,
"step": 141
},
{
"epoch": 1.4343434343434343,
"grad_norm": 0.6525304317474365,
"learning_rate": 1.575757575757576e-05,
"loss": 6.9673,
"step": 142
},
{
"epoch": 1.4444444444444444,
"grad_norm": 0.875150203704834,
"learning_rate": 1.5656565656565655e-05,
"loss": 6.9106,
"step": 143
},
{
"epoch": 1.4545454545454546,
"grad_norm": 0.7586594223976135,
"learning_rate": 1.5555555555555555e-05,
"loss": 6.9954,
"step": 144
},
{
"epoch": 1.4646464646464645,
"grad_norm": 1.1310893297195435,
"learning_rate": 1.5454545454545454e-05,
"loss": 7.1989,
"step": 145
},
{
"epoch": 1.4747474747474747,
"grad_norm": 1.3009579181671143,
"learning_rate": 1.5353535353535354e-05,
"loss": 7.1353,
"step": 146
},
{
"epoch": 1.4848484848484849,
"grad_norm": 0.6209822297096252,
"learning_rate": 1.5252525252525253e-05,
"loss": 6.9722,
"step": 147
},
{
"epoch": 1.494949494949495,
"grad_norm": 0.7535139322280884,
"learning_rate": 1.5151515151515153e-05,
"loss": 7.0278,
"step": 148
},
{
"epoch": 1.5050505050505052,
"grad_norm": 0.6947258114814758,
"learning_rate": 1.5050505050505052e-05,
"loss": 7.0663,
"step": 149
},
{
"epoch": 1.5151515151515151,
"grad_norm": 0.7372407913208008,
"learning_rate": 1.494949494949495e-05,
"loss": 6.9891,
"step": 150
},
{
"epoch": 1.5252525252525253,
"grad_norm": 0.8788199424743652,
"learning_rate": 1.484848484848485e-05,
"loss": 6.9849,
"step": 151
},
{
"epoch": 1.5353535353535355,
"grad_norm": 0.9445679187774658,
"learning_rate": 1.4747474747474747e-05,
"loss": 6.976,
"step": 152
},
{
"epoch": 1.5454545454545454,
"grad_norm": 1.7923697233200073,
"learning_rate": 1.4646464646464647e-05,
"loss": 6.7889,
"step": 153
},
{
"epoch": 1.5555555555555556,
"grad_norm": 0.6855000257492065,
"learning_rate": 1.4545454545454546e-05,
"loss": 6.9884,
"step": 154
},
{
"epoch": 1.5656565656565657,
"grad_norm": 0.8418505191802979,
"learning_rate": 1.4444444444444444e-05,
"loss": 6.9349,
"step": 155
},
{
"epoch": 1.5757575757575757,
"grad_norm": 0.7767158150672913,
"learning_rate": 1.4343434343434344e-05,
"loss": 6.9518,
"step": 156
},
{
"epoch": 1.5858585858585859,
"grad_norm": 0.9679882526397705,
"learning_rate": 1.4242424242424243e-05,
"loss": 6.8211,
"step": 157
},
{
"epoch": 1.595959595959596,
"grad_norm": 0.9250569343566895,
"learning_rate": 1.4141414141414143e-05,
"loss": 6.9044,
"step": 158
},
{
"epoch": 1.606060606060606,
"grad_norm": 0.8403336405754089,
"learning_rate": 1.404040404040404e-05,
"loss": 6.8393,
"step": 159
},
{
"epoch": 1.6161616161616161,
"grad_norm": 0.9166443347930908,
"learning_rate": 1.393939393939394e-05,
"loss": 6.8174,
"step": 160
},
{
"epoch": 1.6262626262626263,
"grad_norm": 0.5435317754745483,
"learning_rate": 1.383838383838384e-05,
"loss": 6.9202,
"step": 161
},
{
"epoch": 1.6363636363636362,
"grad_norm": 1.0361636877059937,
"learning_rate": 1.3737373737373739e-05,
"loss": 6.9836,
"step": 162
},
{
"epoch": 1.6464646464646466,
"grad_norm": 1.709415316581726,
"learning_rate": 1.3636363636363637e-05,
"loss": 7.1939,
"step": 163
},
{
"epoch": 1.6565656565656566,
"grad_norm": 1.3511853218078613,
"learning_rate": 1.3535353535353536e-05,
"loss": 7.0895,
"step": 164
},
{
"epoch": 1.6666666666666665,
"grad_norm": 0.6736574172973633,
"learning_rate": 1.3434343434343436e-05,
"loss": 6.8909,
"step": 165
},
{
"epoch": 1.676767676767677,
"grad_norm": 0.6862733364105225,
"learning_rate": 1.3333333333333333e-05,
"loss": 6.9872,
"step": 166
},
{
"epoch": 1.6868686868686869,
"grad_norm": 1.0613605976104736,
"learning_rate": 1.3232323232323233e-05,
"loss": 6.9787,
"step": 167
},
{
"epoch": 1.696969696969697,
"grad_norm": 0.9397658705711365,
"learning_rate": 1.3131313131313132e-05,
"loss": 6.85,
"step": 168
},
{
"epoch": 1.7070707070707072,
"grad_norm": 0.6116166114807129,
"learning_rate": 1.3030303030303032e-05,
"loss": 6.9257,
"step": 169
},
{
"epoch": 1.7171717171717171,
"grad_norm": 1.0234121084213257,
"learning_rate": 1.292929292929293e-05,
"loss": 7.0316,
"step": 170
},
{
"epoch": 1.7272727272727273,
"grad_norm": 1.68846595287323,
"learning_rate": 1.2828282828282829e-05,
"loss": 7.1854,
"step": 171
},
{
"epoch": 1.7373737373737375,
"grad_norm": 0.686038076877594,
"learning_rate": 1.2727272727272728e-05,
"loss": 6.9311,
"step": 172
},
{
"epoch": 1.7474747474747474,
"grad_norm": 1.3312983512878418,
"learning_rate": 1.2626262626262628e-05,
"loss": 7.1633,
"step": 173
},
{
"epoch": 1.7575757575757576,
"grad_norm": 0.688218355178833,
"learning_rate": 1.2525252525252526e-05,
"loss": 7.0336,
"step": 174
},
{
"epoch": 1.7676767676767677,
"grad_norm": 0.768675684928894,
"learning_rate": 1.2424242424242425e-05,
"loss": 6.9246,
"step": 175
},
{
"epoch": 1.7777777777777777,
"grad_norm": 0.6171760559082031,
"learning_rate": 1.2323232323232325e-05,
"loss": 7.0127,
"step": 176
},
{
"epoch": 1.7878787878787878,
"grad_norm": 0.79632967710495,
"learning_rate": 1.2222222222222222e-05,
"loss": 7.0955,
"step": 177
},
{
"epoch": 1.797979797979798,
"grad_norm": 0.5743001699447632,
"learning_rate": 1.2121212121212122e-05,
"loss": 7.0305,
"step": 178
},
{
"epoch": 1.808080808080808,
"grad_norm": 0.648186445236206,
"learning_rate": 1.2020202020202021e-05,
"loss": 7.0791,
"step": 179
},
{
"epoch": 1.8181818181818183,
"grad_norm": 0.9388221502304077,
"learning_rate": 1.1919191919191921e-05,
"loss": 6.8963,
"step": 180
},
{
"epoch": 1.8282828282828283,
"grad_norm": 0.6248198747634888,
"learning_rate": 1.1818181818181819e-05,
"loss": 7.1456,
"step": 181
},
{
"epoch": 1.8383838383838382,
"grad_norm": 0.7853441834449768,
"learning_rate": 1.1717171717171718e-05,
"loss": 7.1302,
"step": 182
},
{
"epoch": 1.8484848484848486,
"grad_norm": 0.7538427710533142,
"learning_rate": 1.1616161616161618e-05,
"loss": 7.0855,
"step": 183
},
{
"epoch": 1.8585858585858586,
"grad_norm": 1.3595892190933228,
"learning_rate": 1.1515151515151515e-05,
"loss": 6.8817,
"step": 184
},
{
"epoch": 1.8686868686868687,
"grad_norm": 1.2711068391799927,
"learning_rate": 1.1414141414141415e-05,
"loss": 6.9406,
"step": 185
},
{
"epoch": 1.878787878787879,
"grad_norm": 0.7485206127166748,
"learning_rate": 1.1313131313131314e-05,
"loss": 7.0214,
"step": 186
},
{
"epoch": 1.8888888888888888,
"grad_norm": 1.068521499633789,
"learning_rate": 1.1212121212121212e-05,
"loss": 6.9544,
"step": 187
},
{
"epoch": 1.898989898989899,
"grad_norm": 0.6432831287384033,
"learning_rate": 1.111111111111111e-05,
"loss": 7.0268,
"step": 188
},
{
"epoch": 1.9090909090909092,
"grad_norm": 1.0608913898468018,
"learning_rate": 1.101010101010101e-05,
"loss": 6.9549,
"step": 189
},
{
"epoch": 1.9191919191919191,
"grad_norm": 0.7320489883422852,
"learning_rate": 1.0909090909090909e-05,
"loss": 6.9866,
"step": 190
},
{
"epoch": 1.9292929292929293,
"grad_norm": 1.776832103729248,
"learning_rate": 1.0808080808080808e-05,
"loss": 6.8107,
"step": 191
},
{
"epoch": 1.9393939393939394,
"grad_norm": 0.8337501883506775,
"learning_rate": 1.0707070707070706e-05,
"loss": 6.9117,
"step": 192
},
{
"epoch": 1.9494949494949494,
"grad_norm": 0.9856388568878174,
"learning_rate": 1.0606060606060606e-05,
"loss": 6.8311,
"step": 193
},
{
"epoch": 1.9595959595959596,
"grad_norm": 1.3855446577072144,
"learning_rate": 1.0505050505050505e-05,
"loss": 7.1336,
"step": 194
},
{
"epoch": 1.9696969696969697,
"grad_norm": 1.5992196798324585,
"learning_rate": 1.0404040404040403e-05,
"loss": 6.6699,
"step": 195
},
{
"epoch": 1.9797979797979797,
"grad_norm": 2.3153018951416016,
"learning_rate": 1.0303030303030302e-05,
"loss": 7.2326,
"step": 196
},
{
"epoch": 1.98989898989899,
"grad_norm": 0.5802290439605713,
"learning_rate": 1.0202020202020202e-05,
"loss": 6.9314,
"step": 197
},
{
"epoch": 2.0,
"grad_norm": 0.8023201823234558,
"learning_rate": 1.0101010101010101e-05,
"loss": 6.7947,
"step": 198
},
{
"epoch": 2.0,
"eval_audio_cosine_sim": 0.37472105026245117,
"eval_loss": 5.077270030975342,
"eval_runtime": 152.9953,
"eval_samples_per_second": 0.059,
"eval_steps_per_second": 0.02,
"eval_text_cosine_sim": 0.1508849412202835,
"step": 198
},
{
"epoch": 2.01010101010101,
"grad_norm": 0.8875887989997864,
"learning_rate": 9.999999999999999e-06,
"loss": 6.9461,
"step": 199
},
{
"epoch": 2.0202020202020203,
"grad_norm": 0.9611765742301941,
"learning_rate": 9.898989898989899e-06,
"loss": 6.9659,
"step": 200
},
{
"epoch": 2.0303030303030303,
"grad_norm": 1.539200782775879,
"learning_rate": 9.797979797979798e-06,
"loss": 6.6096,
"step": 201
},
{
"epoch": 2.04040404040404,
"grad_norm": 0.8153275847434998,
"learning_rate": 9.696969696969698e-06,
"loss": 6.8189,
"step": 202
},
{
"epoch": 2.0505050505050506,
"grad_norm": 0.8883192539215088,
"learning_rate": 9.595959595959595e-06,
"loss": 6.8387,
"step": 203
},
{
"epoch": 2.0606060606060606,
"grad_norm": 1.5102068185806274,
"learning_rate": 9.494949494949495e-06,
"loss": 6.5686,
"step": 204
},
{
"epoch": 2.0707070707070705,
"grad_norm": 0.86061692237854,
"learning_rate": 9.393939393939394e-06,
"loss": 6.8586,
"step": 205
},
{
"epoch": 2.080808080808081,
"grad_norm": 1.1872936487197876,
"learning_rate": 9.292929292929292e-06,
"loss": 6.9541,
"step": 206
},
{
"epoch": 2.090909090909091,
"grad_norm": 0.6017976999282837,
"learning_rate": 9.191919191919192e-06,
"loss": 6.7969,
"step": 207
},
{
"epoch": 2.101010101010101,
"grad_norm": 2.1137189865112305,
"learning_rate": 9.090909090909091e-06,
"loss": 7.2209,
"step": 208
},
{
"epoch": 2.111111111111111,
"grad_norm": 0.7061384916305542,
"learning_rate": 8.98989898989899e-06,
"loss": 6.7687,
"step": 209
},
{
"epoch": 2.121212121212121,
"grad_norm": 0.7401531934738159,
"learning_rate": 8.888888888888888e-06,
"loss": 6.8944,
"step": 210
},
{
"epoch": 2.1313131313131315,
"grad_norm": 1.1122565269470215,
"learning_rate": 8.787878787878788e-06,
"loss": 6.6628,
"step": 211
},
{
"epoch": 2.1414141414141414,
"grad_norm": 0.6532828211784363,
"learning_rate": 8.686868686868687e-06,
"loss": 6.8288,
"step": 212
},
{
"epoch": 2.1515151515151514,
"grad_norm": 1.3121882677078247,
"learning_rate": 8.585858585858585e-06,
"loss": 7.0173,
"step": 213
},
{
"epoch": 2.1616161616161618,
"grad_norm": 1.4856034517288208,
"learning_rate": 8.484848484848485e-06,
"loss": 7.0473,
"step": 214
},
{
"epoch": 2.1717171717171717,
"grad_norm": 1.0639023780822754,
"learning_rate": 8.383838383838384e-06,
"loss": 6.9259,
"step": 215
},
{
"epoch": 2.1818181818181817,
"grad_norm": 0.9515113830566406,
"learning_rate": 8.282828282828283e-06,
"loss": 6.9661,
"step": 216
},
{
"epoch": 2.191919191919192,
"grad_norm": 0.6862073540687561,
"learning_rate": 8.181818181818181e-06,
"loss": 6.7838,
"step": 217
},
{
"epoch": 2.202020202020202,
"grad_norm": 0.6430158615112305,
"learning_rate": 8.08080808080808e-06,
"loss": 6.9264,
"step": 218
},
{
"epoch": 2.212121212121212,
"grad_norm": 0.9119675159454346,
"learning_rate": 7.97979797979798e-06,
"loss": 6.9994,
"step": 219
},
{
"epoch": 2.2222222222222223,
"grad_norm": 1.3677645921707153,
"learning_rate": 7.87878787878788e-06,
"loss": 6.6887,
"step": 220
},
{
"epoch": 2.2323232323232323,
"grad_norm": 1.3011837005615234,
"learning_rate": 7.777777777777777e-06,
"loss": 6.6735,
"step": 221
},
{
"epoch": 2.242424242424242,
"grad_norm": 0.8368801474571228,
"learning_rate": 7.676767676767677e-06,
"loss": 6.9868,
"step": 222
},
{
"epoch": 2.2525252525252526,
"grad_norm": 1.2124541997909546,
"learning_rate": 7.5757575757575764e-06,
"loss": 7.0907,
"step": 223
},
{
"epoch": 2.2626262626262625,
"grad_norm": 0.5335760116577148,
"learning_rate": 7.474747474747475e-06,
"loss": 6.9152,
"step": 224
},
{
"epoch": 2.2727272727272725,
"grad_norm": 1.2292662858963013,
"learning_rate": 7.373737373737374e-06,
"loss": 7.0845,
"step": 225
},
{
"epoch": 2.282828282828283,
"grad_norm": 1.1617155075073242,
"learning_rate": 7.272727272727273e-06,
"loss": 6.8596,
"step": 226
},
{
"epoch": 2.292929292929293,
"grad_norm": 1.7857162952423096,
"learning_rate": 7.171717171717172e-06,
"loss": 6.7016,
"step": 227
},
{
"epoch": 2.303030303030303,
"grad_norm": 0.7340604066848755,
"learning_rate": 7.070707070707071e-06,
"loss": 6.924,
"step": 228
},
{
"epoch": 2.313131313131313,
"grad_norm": 1.880449652671814,
"learning_rate": 6.96969696969697e-06,
"loss": 7.1343,
"step": 229
},
{
"epoch": 2.323232323232323,
"grad_norm": 0.9435675144195557,
"learning_rate": 6.868686868686869e-06,
"loss": 6.8979,
"step": 230
},
{
"epoch": 2.3333333333333335,
"grad_norm": 0.877120316028595,
"learning_rate": 6.767676767676768e-06,
"loss": 6.8256,
"step": 231
},
{
"epoch": 2.3434343434343434,
"grad_norm": 0.9317502975463867,
"learning_rate": 6.666666666666667e-06,
"loss": 6.8409,
"step": 232
},
{
"epoch": 2.3535353535353534,
"grad_norm": 1.0060955286026,
"learning_rate": 6.565656565656566e-06,
"loss": 7.0845,
"step": 233
},
{
"epoch": 2.3636363636363638,
"grad_norm": 0.8277117609977722,
"learning_rate": 6.464646464646465e-06,
"loss": 7.092,
"step": 234
},
{
"epoch": 2.3737373737373737,
"grad_norm": 0.755837619304657,
"learning_rate": 6.363636363636364e-06,
"loss": 6.8859,
"step": 235
},
{
"epoch": 2.3838383838383836,
"grad_norm": 0.6454436182975769,
"learning_rate": 6.262626262626263e-06,
"loss": 7.0516,
"step": 236
},
{
"epoch": 2.393939393939394,
"grad_norm": 1.4383716583251953,
"learning_rate": 6.161616161616162e-06,
"loss": 6.7176,
"step": 237
},
{
"epoch": 2.404040404040404,
"grad_norm": 1.2009567022323608,
"learning_rate": 6.060606060606061e-06,
"loss": 6.9433,
"step": 238
},
{
"epoch": 2.4141414141414144,
"grad_norm": 0.7479004859924316,
"learning_rate": 5.9595959595959605e-06,
"loss": 6.9019,
"step": 239
},
{
"epoch": 2.4242424242424243,
"grad_norm": 0.6840262413024902,
"learning_rate": 5.858585858585859e-06,
"loss": 6.9329,
"step": 240
},
{
"epoch": 2.4343434343434343,
"grad_norm": 0.6570662260055542,
"learning_rate": 5.757575757575758e-06,
"loss": 6.9839,
"step": 241
},
{
"epoch": 2.4444444444444446,
"grad_norm": 1.0435160398483276,
"learning_rate": 5.656565656565657e-06,
"loss": 6.9886,
"step": 242
},
{
"epoch": 2.4545454545454546,
"grad_norm": 1.2880451679229736,
"learning_rate": 5.555555555555555e-06,
"loss": 7.0024,
"step": 243
},
{
"epoch": 2.4646464646464645,
"grad_norm": 1.2956496477127075,
"learning_rate": 5.4545454545454545e-06,
"loss": 7.1377,
"step": 244
},
{
"epoch": 2.474747474747475,
"grad_norm": 1.3433921337127686,
"learning_rate": 5.353535353535353e-06,
"loss": 7.0797,
"step": 245
},
{
"epoch": 2.484848484848485,
"grad_norm": 0.6715502738952637,
"learning_rate": 5.2525252525252526e-06,
"loss": 6.8772,
"step": 246
},
{
"epoch": 2.494949494949495,
"grad_norm": 0.8352294564247131,
"learning_rate": 5.151515151515151e-06,
"loss": 7.0575,
"step": 247
},
{
"epoch": 2.505050505050505,
"grad_norm": 1.1144355535507202,
"learning_rate": 5.050505050505051e-06,
"loss": 6.8229,
"step": 248
},
{
"epoch": 2.515151515151515,
"grad_norm": 0.6648267507553101,
"learning_rate": 4.949494949494949e-06,
"loss": 7.0238,
"step": 249
},
{
"epoch": 2.525252525252525,
"grad_norm": 0.7869075536727905,
"learning_rate": 4.848484848484849e-06,
"loss": 7.0076,
"step": 250
},
{
"epoch": 2.5353535353535355,
"grad_norm": 0.6390742659568787,
"learning_rate": 4.747474747474747e-06,
"loss": 7.0491,
"step": 251
},
{
"epoch": 2.5454545454545454,
"grad_norm": 0.7328326106071472,
"learning_rate": 4.646464646464646e-06,
"loss": 7.0393,
"step": 252
},
{
"epoch": 2.5555555555555554,
"grad_norm": 0.8217814564704895,
"learning_rate": 4.5454545454545455e-06,
"loss": 6.9342,
"step": 253
},
{
"epoch": 2.5656565656565657,
"grad_norm": 1.3768435716629028,
"learning_rate": 4.444444444444444e-06,
"loss": 7.124,
"step": 254
},
{
"epoch": 2.5757575757575757,
"grad_norm": 1.0091509819030762,
"learning_rate": 4.343434343434344e-06,
"loss": 6.8746,
"step": 255
},
{
"epoch": 2.5858585858585856,
"grad_norm": 1.0249191522598267,
"learning_rate": 4.242424242424242e-06,
"loss": 6.8636,
"step": 256
},
{
"epoch": 2.595959595959596,
"grad_norm": 0.9580966234207153,
"learning_rate": 4.141414141414142e-06,
"loss": 6.9923,
"step": 257
},
{
"epoch": 2.606060606060606,
"grad_norm": 1.5487310886383057,
"learning_rate": 4.04040404040404e-06,
"loss": 6.8229,
"step": 258
},
{
"epoch": 2.616161616161616,
"grad_norm": 0.6319442987442017,
"learning_rate": 3.93939393939394e-06,
"loss": 7.0224,
"step": 259
},
{
"epoch": 2.6262626262626263,
"grad_norm": 0.7614464163780212,
"learning_rate": 3.8383838383838385e-06,
"loss": 6.9499,
"step": 260
},
{
"epoch": 2.6363636363636362,
"grad_norm": 0.9656277298927307,
"learning_rate": 3.7373737373737375e-06,
"loss": 6.881,
"step": 261
},
{
"epoch": 2.6464646464646466,
"grad_norm": 2.4072742462158203,
"learning_rate": 3.6363636363636366e-06,
"loss": 6.579,
"step": 262
},
{
"epoch": 2.6565656565656566,
"grad_norm": 1.361374855041504,
"learning_rate": 3.5353535353535356e-06,
"loss": 7.1662,
"step": 263
},
{
"epoch": 2.6666666666666665,
"grad_norm": 0.7471557855606079,
"learning_rate": 3.4343434343434347e-06,
"loss": 7.026,
"step": 264
},
{
"epoch": 2.676767676767677,
"grad_norm": 0.9399821758270264,
"learning_rate": 3.3333333333333333e-06,
"loss": 6.8437,
"step": 265
},
{
"epoch": 2.686868686868687,
"grad_norm": 0.960325300693512,
"learning_rate": 3.2323232323232324e-06,
"loss": 7.1221,
"step": 266
},
{
"epoch": 2.6969696969696972,
"grad_norm": 0.7455516457557678,
"learning_rate": 3.1313131313131314e-06,
"loss": 6.8993,
"step": 267
},
{
"epoch": 2.707070707070707,
"grad_norm": 0.7760592103004456,
"learning_rate": 3.0303030303030305e-06,
"loss": 6.9543,
"step": 268
},
{
"epoch": 2.717171717171717,
"grad_norm": 0.7888494729995728,
"learning_rate": 2.9292929292929295e-06,
"loss": 7.0662,
"step": 269
},
{
"epoch": 2.7272727272727275,
"grad_norm": 1.3352677822113037,
"learning_rate": 2.8282828282828286e-06,
"loss": 6.801,
"step": 270
},
{
"epoch": 2.7373737373737375,
"grad_norm": 0.7915419936180115,
"learning_rate": 2.7272727272727272e-06,
"loss": 7.0028,
"step": 271
},
{
"epoch": 2.7474747474747474,
"grad_norm": 0.5938275456428528,
"learning_rate": 2.6262626262626263e-06,
"loss": 6.9275,
"step": 272
},
{
"epoch": 2.757575757575758,
"grad_norm": 1.24167001247406,
"learning_rate": 2.5252525252525253e-06,
"loss": 6.7646,
"step": 273
},
{
"epoch": 2.7676767676767677,
"grad_norm": 1.3691089153289795,
"learning_rate": 2.4242424242424244e-06,
"loss": 7.0922,
"step": 274
},
{
"epoch": 2.7777777777777777,
"grad_norm": 1.1850454807281494,
"learning_rate": 2.323232323232323e-06,
"loss": 7.0915,
"step": 275
},
{
"epoch": 2.787878787878788,
"grad_norm": 1.693428874015808,
"learning_rate": 2.222222222222222e-06,
"loss": 6.7004,
"step": 276
},
{
"epoch": 2.797979797979798,
"grad_norm": 0.5935156345367432,
"learning_rate": 2.121212121212121e-06,
"loss": 6.9309,
"step": 277
},
{
"epoch": 2.808080808080808,
"grad_norm": 0.6563141942024231,
"learning_rate": 2.02020202020202e-06,
"loss": 7.0097,
"step": 278
},
{
"epoch": 2.8181818181818183,
"grad_norm": 0.9683242440223694,
"learning_rate": 1.9191919191919192e-06,
"loss": 7.0719,
"step": 279
},
{
"epoch": 2.8282828282828283,
"grad_norm": 1.0628868341445923,
"learning_rate": 1.8181818181818183e-06,
"loss": 7.0084,
"step": 280
},
{
"epoch": 2.8383838383838382,
"grad_norm": 0.7137049436569214,
"learning_rate": 1.7171717171717173e-06,
"loss": 6.9748,
"step": 281
},
{
"epoch": 2.8484848484848486,
"grad_norm": 1.1555520296096802,
"learning_rate": 1.6161616161616162e-06,
"loss": 6.8182,
"step": 282
},
{
"epoch": 2.8585858585858586,
"grad_norm": 1.3852341175079346,
"learning_rate": 1.5151515151515152e-06,
"loss": 7.1022,
"step": 283
},
{
"epoch": 2.8686868686868685,
"grad_norm": 0.6663222908973694,
"learning_rate": 1.4141414141414143e-06,
"loss": 7.0444,
"step": 284
},
{
"epoch": 2.878787878787879,
"grad_norm": 0.9903120994567871,
"learning_rate": 1.3131313131313131e-06,
"loss": 6.7819,
"step": 285
},
{
"epoch": 2.888888888888889,
"grad_norm": 1.510622262954712,
"learning_rate": 1.2121212121212122e-06,
"loss": 7.1841,
"step": 286
},
{
"epoch": 2.898989898989899,
"grad_norm": 1.2110823392868042,
"learning_rate": 1.111111111111111e-06,
"loss": 7.0199,
"step": 287
},
{
"epoch": 2.909090909090909,
"grad_norm": 1.0056835412979126,
"learning_rate": 1.01010101010101e-06,
"loss": 6.9958,
"step": 288
},
{
"epoch": 2.919191919191919,
"grad_norm": 0.8939002156257629,
"learning_rate": 9.090909090909091e-07,
"loss": 7.0829,
"step": 289
},
{
"epoch": 2.929292929292929,
"grad_norm": 0.6934316158294678,
"learning_rate": 8.080808080808081e-07,
"loss": 7.015,
"step": 290
},
{
"epoch": 2.9393939393939394,
"grad_norm": 0.7522878050804138,
"learning_rate": 7.070707070707071e-07,
"loss": 7.0191,
"step": 291
},
{
"epoch": 2.9494949494949494,
"grad_norm": 1.2762123346328735,
"learning_rate": 6.060606060606061e-07,
"loss": 6.7045,
"step": 292
},
{
"epoch": 2.9595959595959593,
"grad_norm": 1.0797396898269653,
"learning_rate": 5.05050505050505e-07,
"loss": 7.0543,
"step": 293
},
{
"epoch": 2.9696969696969697,
"grad_norm": 1.322259783744812,
"learning_rate": 4.0404040404040405e-07,
"loss": 6.7815,
"step": 294
},
{
"epoch": 2.9797979797979797,
"grad_norm": 0.63960200548172,
"learning_rate": 3.0303030303030305e-07,
"loss": 6.9333,
"step": 295
},
{
"epoch": 2.98989898989899,
"grad_norm": 0.8398324847221375,
"learning_rate": 2.0202020202020202e-07,
"loss": 7.0499,
"step": 296
},
{
"epoch": 3.0,
"grad_norm": 1.3194384574890137,
"learning_rate": 1.0101010101010101e-07,
"loss": 7.1508,
"step": 297
},
{
"epoch": 3.0,
"eval_audio_cosine_sim": 0.32165560126304626,
"eval_loss": 5.101603984832764,
"eval_runtime": 153.0389,
"eval_samples_per_second": 0.059,
"eval_steps_per_second": 0.02,
"eval_text_cosine_sim": 0.11808153986930847,
"step": 297
},
{
"epoch": 3.0,
"step": 297,
"total_flos": 6235958582540448.0,
"train_loss": 7.227042491989907,
"train_runtime": 3530.6424,
"train_samples_per_second": 1.343,
"train_steps_per_second": 0.084
}
],
"logging_steps": 1.0,
"max_steps": 297,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": false,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 6235958582540448.0,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}