End of training

README.md

@@ -2,6 +2,8 @@
 license: cc-by-nc-4.0
 library_name: peft
 tags:
+- text-to-audio
+- sweet-dreambooths/combined-artists-v1
 - generated_from_trainer
 base_model: facebook/musicgen-melody-large
 model-index:
@@ -16,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/sanchit-gandhi/black-eyed-peas/runs/aot7xf5c)
 # combined-artists-text-only-1-epochs
 
-This model is a fine-tuned version of [facebook/musicgen-melody-large](https://huggingface.co/facebook/musicgen-melody-large) on an unknown dataset.
+This model is a fine-tuned version of [facebook/musicgen-melody-large](https://huggingface.co/facebook/musicgen-melody-large) on the SWEET-DREAMBOOTHS/COMBINED-ARTISTS-V1 - DEFAULT dataset.
 It achieves the following results on the evaluation set:
 - Loss: 3.9133
-- Text Cosine Sim: 0.2471
-- Audio Cosine Sim: 0.4541
+- Text Cosine Sim: 0.2331
+- Audio Cosine Sim: 0.4680
 
 ## Model description
 

all_results.json

@@ -0,0 +1,16 @@
+{
+    "epoch": 1.0,
+    "eval_audio_cosine_sim": 0.4679793119430542,
+    "eval_loss": 3.913311243057251,
+    "eval_runtime": 237.6952,
+    "eval_samples": 9,
+    "eval_samples_per_second": 0.038,
+    "eval_steps_per_second": 0.013,
+    "eval_text_cosine_sim": 0.23309490084648132,
+    "total_flos": 2077734266890272.0,
+    "train_loss": 7.893082406785753,
+    "train_runtime": 1993.2394,
+    "train_samples": 1580,
+    "train_samples_per_second": 0.793,
+    "train_steps_per_second": 0.05
+}

eval_results.json

@@ -0,0 +1,10 @@
+{
+    "epoch": 1.0,
+    "eval_audio_cosine_sim": 0.4679793119430542,
+    "eval_loss": 3.913311243057251,
+    "eval_runtime": 237.6952,
+    "eval_samples": 9,
+    "eval_samples_per_second": 0.038,
+    "eval_steps_per_second": 0.013,
+    "eval_text_cosine_sim": 0.23309490084648132
+}

train_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 1.0,
+    "total_flos": 2077734266890272.0,
+    "train_loss": 7.893082406785753,
+    "train_runtime": 1993.2394,
+    "train_samples": 1580,
+    "train_samples_per_second": 0.793,
+    "train_steps_per_second": 0.05
+}

trainer_state.json

@@ -0,0 +1,745 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 99,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.010101010101010102,
+      "grad_norm": 1.4316415786743164,
+      "learning_rate": 2.96969696969697e-05,
+      "loss": 10.0726,
+      "step": 1
+    },
+    {
+      "epoch": 0.020202020202020204,
+      "grad_norm": 1.560971736907959,
+      "learning_rate": 2.9393939393939394e-05,
+      "loss": 10.129,
+      "step": 2
+    },
+    {
+      "epoch": 0.030303030303030304,
+      "grad_norm": 1.4093116521835327,
+      "learning_rate": 2.9090909090909093e-05,
+      "loss": 10.0404,
+      "step": 3
+    },
+    {
+      "epoch": 0.04040404040404041,
+      "grad_norm": 1.5073634386062622,
+      "learning_rate": 2.8787878787878788e-05,
+      "loss": 10.0214,
+      "step": 4
+    },
+    {
+      "epoch": 0.050505050505050504,
+      "grad_norm": 1.6120322942733765,
+      "learning_rate": 2.8484848484848486e-05,
+      "loss": 10.0287,
+      "step": 5
+    },
+    {
+      "epoch": 0.06060606060606061,
+      "grad_norm": 1.8709332942962646,
+      "learning_rate": 2.8181818181818185e-05,
+      "loss": 10.0744,
+      "step": 6
+    },
+    {
+      "epoch": 0.0707070707070707,
+      "grad_norm": 1.8668410778045654,
+      "learning_rate": 2.787878787878788e-05,
+      "loss": 9.9539,
+      "step": 7
+    },
+    {
+      "epoch": 0.08080808080808081,
+      "grad_norm": 1.9248428344726562,
+      "learning_rate": 2.7575757575757578e-05,
+      "loss": 9.8166,
+      "step": 8
+    },
+    {
+      "epoch": 0.09090909090909091,
+      "grad_norm": 1.9779285192489624,
+      "learning_rate": 2.7272727272727273e-05,
+      "loss": 9.7811,
+      "step": 9
+    },
+    {
+      "epoch": 0.10101010101010101,
+      "grad_norm": 2.0661659240722656,
+      "learning_rate": 2.696969696969697e-05,
+      "loss": 9.743,
+      "step": 10
+    },
+    {
+      "epoch": 0.1111111111111111,
+      "grad_norm": 2.304006338119507,
+      "learning_rate": 2.6666666666666667e-05,
+      "loss": 9.7368,
+      "step": 11
+    },
+    {
+      "epoch": 0.12121212121212122,
+      "grad_norm": 2.2433547973632812,
+      "learning_rate": 2.6363636363636365e-05,
+      "loss": 9.6052,
+      "step": 12
+    },
+    {
+      "epoch": 0.13131313131313133,
+      "grad_norm": 2.3963887691497803,
+      "learning_rate": 2.6060606060606063e-05,
+      "loss": 9.5241,
+      "step": 13
+    },
+    {
+      "epoch": 0.1414141414141414,
+      "grad_norm": 2.690804958343506,
+      "learning_rate": 2.575757575757576e-05,
+      "loss": 9.4771,
+      "step": 14
+    },
+    {
+      "epoch": 0.15151515151515152,
+      "grad_norm": 2.54472017288208,
+      "learning_rate": 2.5454545454545457e-05,
+      "loss": 9.3078,
+      "step": 15
+    },
+    {
+      "epoch": 0.16161616161616163,
+      "grad_norm": 2.672468662261963,
+      "learning_rate": 2.5151515151515152e-05,
+      "loss": 9.295,
+      "step": 16
+    },
+    {
+      "epoch": 0.1717171717171717,
+      "grad_norm": 2.9068398475646973,
+      "learning_rate": 2.484848484848485e-05,
+      "loss": 9.233,
+      "step": 17
+    },
+    {
+      "epoch": 0.18181818181818182,
+      "grad_norm": 2.6378564834594727,
+      "learning_rate": 2.454545454545455e-05,
+      "loss": 9.0833,
+      "step": 18
+    },
+    {
+      "epoch": 0.1919191919191919,
+      "grad_norm": 2.9194931983947754,
+      "learning_rate": 2.4242424242424244e-05,
+      "loss": 9.0367,
+      "step": 19
+    },
+    {
+      "epoch": 0.20202020202020202,
+      "grad_norm": 2.9216768741607666,
+      "learning_rate": 2.3939393939393942e-05,
+      "loss": 8.9809,
+      "step": 20
+    },
+    {
+      "epoch": 0.21212121212121213,
+      "grad_norm": 2.6998298168182373,
+      "learning_rate": 2.3636363636363637e-05,
+      "loss": 8.8696,
+      "step": 21
+    },
+    {
+      "epoch": 0.2222222222222222,
+      "grad_norm": 2.8181002140045166,
+      "learning_rate": 2.3333333333333336e-05,
+      "loss": 8.8195,
+      "step": 22
+    },
+    {
+      "epoch": 0.23232323232323232,
+      "grad_norm": 2.82079815864563,
+      "learning_rate": 2.303030303030303e-05,
+      "loss": 8.7101,
+      "step": 23
+    },
+    {
+      "epoch": 0.24242424242424243,
+      "grad_norm": 2.900256633758545,
+      "learning_rate": 2.272727272727273e-05,
+      "loss": 8.6645,
+      "step": 24
+    },
+    {
+      "epoch": 0.25252525252525254,
+      "grad_norm": 2.9626667499542236,
+      "learning_rate": 2.2424242424242424e-05,
+      "loss": 8.4995,
+      "step": 25
+    },
+    {
+      "epoch": 0.26262626262626265,
+      "grad_norm": 2.78900146484375,
+      "learning_rate": 2.212121212121212e-05,
+      "loss": 8.5197,
+      "step": 26
+    },
+    {
+      "epoch": 0.2727272727272727,
+      "grad_norm": 2.8724753856658936,
+      "learning_rate": 2.1818181818181818e-05,
+      "loss": 8.3911,
+      "step": 27
+    },
+    {
+      "epoch": 0.2828282828282828,
+      "grad_norm": 3.0413928031921387,
+      "learning_rate": 2.1515151515151513e-05,
+      "loss": 8.3371,
+      "step": 28
+    },
+    {
+      "epoch": 0.29292929292929293,
+      "grad_norm": 2.8662936687469482,
+      "learning_rate": 2.121212121212121e-05,
+      "loss": 8.274,
+      "step": 29
+    },
+    {
+      "epoch": 0.30303030303030304,
+      "grad_norm": 2.660316228866577,
+      "learning_rate": 2.090909090909091e-05,
+      "loss": 8.2402,
+      "step": 30
+    },
+    {
+      "epoch": 0.31313131313131315,
+      "grad_norm": 2.8771886825561523,
+      "learning_rate": 2.0606060606060605e-05,
+      "loss": 8.164,
+      "step": 31
+    },
+    {
+      "epoch": 0.32323232323232326,
+      "grad_norm": 2.6009693145751953,
+      "learning_rate": 2.0303030303030303e-05,
+      "loss": 8.1005,
+      "step": 32
+    },
+    {
+      "epoch": 0.3333333333333333,
+      "grad_norm": 2.431004285812378,
+      "learning_rate": 1.9999999999999998e-05,
+      "loss": 8.0474,
+      "step": 33
+    },
+    {
+      "epoch": 0.3434343434343434,
+      "grad_norm": 2.523425817489624,
+      "learning_rate": 1.9696969696969697e-05,
+      "loss": 7.9858,
+      "step": 34
+    },
+    {
+      "epoch": 0.35353535353535354,
+      "grad_norm": 2.4093704223632812,
+      "learning_rate": 1.9393939393939395e-05,
+      "loss": 7.9892,
+      "step": 35
+    },
+    {
+      "epoch": 0.36363636363636365,
+      "grad_norm": 2.5664756298065186,
+      "learning_rate": 1.909090909090909e-05,
+      "loss": 7.8404,
+      "step": 36
+    },
+    {
+      "epoch": 0.37373737373737376,
+      "grad_norm": 1.9647918939590454,
+      "learning_rate": 1.878787878787879e-05,
+      "loss": 7.8655,
+      "step": 37
+    },
+    {
+      "epoch": 0.3838383838383838,
+      "grad_norm": 2.1500051021575928,
+      "learning_rate": 1.8484848484848484e-05,
+      "loss": 7.8058,
+      "step": 38
+    },
+    {
+      "epoch": 0.3939393939393939,
+      "grad_norm": 1.9459105730056763,
+      "learning_rate": 1.8181818181818182e-05,
+      "loss": 7.821,
+      "step": 39
+    },
+    {
+      "epoch": 0.40404040404040403,
+      "grad_norm": 1.8056164979934692,
+      "learning_rate": 1.7878787878787877e-05,
+      "loss": 7.814,
+      "step": 40
+    },
+    {
+      "epoch": 0.41414141414141414,
+      "grad_norm": 2.448673725128174,
+      "learning_rate": 1.7575757575757576e-05,
+      "loss": 7.5915,
+      "step": 41
+    },
+    {
+      "epoch": 0.42424242424242425,
+      "grad_norm": 1.6147074699401855,
+      "learning_rate": 1.7272727272727274e-05,
+      "loss": 7.7663,
+      "step": 42
+    },
+    {
+      "epoch": 0.43434343434343436,
+      "grad_norm": 1.7669254541397095,
+      "learning_rate": 1.696969696969697e-05,
+      "loss": 7.6066,
+      "step": 43
+    },
+    {
+      "epoch": 0.4444444444444444,
+      "grad_norm": 1.938239574432373,
+      "learning_rate": 1.6666666666666667e-05,
+      "loss": 7.6132,
+      "step": 44
+    },
+    {
+      "epoch": 0.45454545454545453,
+      "grad_norm": 2.3109822273254395,
+      "learning_rate": 1.6363636363636363e-05,
+      "loss": 7.4384,
+      "step": 45
+    },
+    {
+      "epoch": 0.46464646464646464,
+      "grad_norm": 1.5837619304656982,
+      "learning_rate": 1.606060606060606e-05,
+      "loss": 7.6066,
+      "step": 46
+    },
+    {
+      "epoch": 0.47474747474747475,
+      "grad_norm": 1.6850427389144897,
+      "learning_rate": 1.575757575757576e-05,
+      "loss": 7.474,
+      "step": 47
+    },
+    {
+      "epoch": 0.48484848484848486,
+      "grad_norm": 1.988224983215332,
+      "learning_rate": 1.5454545454545454e-05,
+      "loss": 7.3918,
+      "step": 48
+    },
+    {
+      "epoch": 0.494949494949495,
+      "grad_norm": 1.754016637802124,
+      "learning_rate": 1.5151515151515153e-05,
+      "loss": 7.3915,
+      "step": 49
+    },
+    {
+      "epoch": 0.5050505050505051,
+      "grad_norm": 2.1182453632354736,
+      "learning_rate": 1.484848484848485e-05,
+      "loss": 7.4276,
+      "step": 50
+    },
+    {
+      "epoch": 0.5151515151515151,
+      "grad_norm": 1.7513837814331055,
+      "learning_rate": 1.4545454545454546e-05,
+      "loss": 7.3634,
+      "step": 51
+    },
+    {
+      "epoch": 0.5252525252525253,
+      "grad_norm": 1.5384173393249512,
+      "learning_rate": 1.4242424242424243e-05,
+      "loss": 7.3046,
+      "step": 52
+    },
+    {
+      "epoch": 0.5353535353535354,
+      "grad_norm": 1.2067221403121948,
+      "learning_rate": 1.393939393939394e-05,
+      "loss": 7.5319,
+      "step": 53
+    },
+    {
+      "epoch": 0.5454545454545454,
+      "grad_norm": 1.1458990573883057,
+      "learning_rate": 1.3636363636363637e-05,
+      "loss": 7.354,
+      "step": 54
+    },
+    {
+      "epoch": 0.5555555555555556,
+      "grad_norm": 1.2268686294555664,
+      "learning_rate": 1.3333333333333333e-05,
+      "loss": 7.3975,
+      "step": 55
+    },
+    {
+      "epoch": 0.5656565656565656,
+      "grad_norm": 1.130756139755249,
+      "learning_rate": 1.3030303030303032e-05,
+      "loss": 7.4055,
+      "step": 56
+    },
+    {
+      "epoch": 0.5757575757575758,
+      "grad_norm": 1.1011919975280762,
+      "learning_rate": 1.2727272727272728e-05,
+      "loss": 7.4282,
+      "step": 57
+    },
+    {
+      "epoch": 0.5858585858585859,
+      "grad_norm": 1.1305310726165771,
+      "learning_rate": 1.2424242424242425e-05,
+      "loss": 7.2983,
+      "step": 58
+    },
+    {
+      "epoch": 0.5959595959595959,
+      "grad_norm": 1.2476590871810913,
+      "learning_rate": 1.2121212121212122e-05,
+      "loss": 7.4395,
+      "step": 59
+    },
+    {
+      "epoch": 0.6060606060606061,
+      "grad_norm": 1.0120925903320312,
+      "learning_rate": 1.1818181818181819e-05,
+      "loss": 7.3328,
+      "step": 60
+    },
+    {
+      "epoch": 0.6161616161616161,
+      "grad_norm": 1.1586886644363403,
+      "learning_rate": 1.1515151515151515e-05,
+      "loss": 7.3609,
+      "step": 61
+    },
+    {
+      "epoch": 0.6262626262626263,
+      "grad_norm": 0.9742720127105713,
+      "learning_rate": 1.1212121212121212e-05,
+      "loss": 7.3204,
+      "step": 62
+    },
+    {
+      "epoch": 0.6363636363636364,
+      "grad_norm": 0.9039322733879089,
+      "learning_rate": 1.0909090909090909e-05,
+      "loss": 7.3262,
+      "step": 63
+    },
+    {
+      "epoch": 0.6464646464646465,
+      "grad_norm": 1.1389368772506714,
+      "learning_rate": 1.0606060606060606e-05,
+      "loss": 7.3952,
+      "step": 64
+    },
+    {
+      "epoch": 0.6565656565656566,
+      "grad_norm": 0.9162051677703857,
+      "learning_rate": 1.0303030303030302e-05,
+      "loss": 7.2239,
+      "step": 65
+    },
+    {
+      "epoch": 0.6666666666666666,
+      "grad_norm": 0.9564353823661804,
+      "learning_rate": 9.999999999999999e-06,
+      "loss": 7.2159,
+      "step": 66
+    },
+    {
+      "epoch": 0.6767676767676768,
+      "grad_norm": 1.0328965187072754,
+      "learning_rate": 9.696969696969698e-06,
+      "loss": 7.2235,
+      "step": 67
+    },
+    {
+      "epoch": 0.6868686868686869,
+      "grad_norm": 1.2637859582901,
+      "learning_rate": 9.393939393939394e-06,
+      "loss": 7.403,
+      "step": 68
+    },
+    {
+      "epoch": 0.696969696969697,
+      "grad_norm": 0.8482282757759094,
+      "learning_rate": 9.090909090909091e-06,
+      "loss": 7.2136,
+      "step": 69
+    },
+    {
+      "epoch": 0.7070707070707071,
+      "grad_norm": 0.748203694820404,
+      "learning_rate": 8.787878787878788e-06,
+      "loss": 7.2005,
+      "step": 70
+    },
+    {
+      "epoch": 0.7171717171717171,
+      "grad_norm": 0.8983027338981628,
+      "learning_rate": 8.484848484848485e-06,
+      "loss": 7.2121,
+      "step": 71
+    },
+    {
+      "epoch": 0.7272727272727273,
+      "grad_norm": 1.0953037738800049,
+      "learning_rate": 8.181818181818181e-06,
+      "loss": 7.2214,
+      "step": 72
+    },
+    {
+      "epoch": 0.7373737373737373,
+      "grad_norm": NaN,
+      "learning_rate": 8.181818181818181e-06,
+      "loss": 0.0,
+      "step": 73
+    },
+    {
+      "epoch": 0.7474747474747475,
+      "grad_norm": 0.7874919772148132,
+      "learning_rate": 7.87878787878788e-06,
+      "loss": 7.2169,
+      "step": 74
+    },
+    {
+      "epoch": 0.7575757575757576,
+      "grad_norm": 1.9311456680297852,
+      "learning_rate": 7.5757575757575764e-06,
+      "loss": 7.1586,
+      "step": 75
+    },
+    {
+      "epoch": 0.7676767676767676,
+      "grad_norm": 0.9535604119300842,
+      "learning_rate": 7.272727272727273e-06,
+      "loss": 7.3104,
+      "step": 76
+    },
+    {
+      "epoch": 0.7777777777777778,
+      "grad_norm": 0.9301349520683289,
+      "learning_rate": 6.96969696969697e-06,
+      "loss": 7.2056,
+      "step": 77
+    },
+    {
+      "epoch": 0.7878787878787878,
+      "grad_norm": 1.5866955518722534,
+      "learning_rate": 6.666666666666667e-06,
+      "loss": 7.1204,
+      "step": 78
+    },
+    {
+      "epoch": 0.797979797979798,
+      "grad_norm": 1.0217430591583252,
+      "learning_rate": 6.363636363636364e-06,
+      "loss": 7.233,
+      "step": 79
+    },
+    {
+      "epoch": 0.8080808080808081,
+      "grad_norm": 0.6997321844100952,
+      "learning_rate": 6.060606060606061e-06,
+      "loss": 7.2076,
+      "step": 80
+    },
+    {
+      "epoch": 0.8181818181818182,
+      "grad_norm": 0.8969782590866089,
+      "learning_rate": 5.757575757575758e-06,
+      "loss": 7.1913,
+      "step": 81
+    },
+    {
+      "epoch": 0.8282828282828283,
+      "grad_norm": 1.0181961059570312,
+      "learning_rate": 5.4545454545454545e-06,
+      "loss": 7.3311,
+      "step": 82
+    },
+    {
+      "epoch": 0.8383838383838383,
+      "grad_norm": 1.2779223918914795,
+      "learning_rate": 5.151515151515151e-06,
+      "loss": 7.0971,
+      "step": 83
+    },
+    {
+      "epoch": 0.8484848484848485,
+      "grad_norm": 0.9610503315925598,
+      "learning_rate": 4.848484848484849e-06,
+      "loss": 7.232,
+      "step": 84
+    },
+    {
+      "epoch": 0.8585858585858586,
+      "grad_norm": 0.8355559706687927,
+      "learning_rate": 4.5454545454545455e-06,
+      "loss": 7.2681,
+      "step": 85
+    },
+    {
+      "epoch": 0.8686868686868687,
+      "grad_norm": 1.0366374254226685,
+      "learning_rate": 4.242424242424242e-06,
+      "loss": 7.2509,
+      "step": 86
+    },
+    {
+      "epoch": 0.8787878787878788,
+      "grad_norm": 0.8738848567008972,
+      "learning_rate": 3.93939393939394e-06,
+      "loss": 7.1105,
+      "step": 87
+    },
+    {
+      "epoch": 0.8888888888888888,
+      "grad_norm": 0.8608594536781311,
+      "learning_rate": 3.6363636363636366e-06,
+      "loss": 7.1243,
+      "step": 88
+    },
+    {
+      "epoch": 0.898989898989899,
+      "grad_norm": 1.1317100524902344,
+      "learning_rate": 3.3333333333333333e-06,
+      "loss": 7.0372,
+      "step": 89
+    },
+    {
+      "epoch": 0.9090909090909091,
+      "grad_norm": 0.802678108215332,
+      "learning_rate": 3.0303030303030305e-06,
+      "loss": 7.1456,
+      "step": 90
+    },
+    {
+      "epoch": 0.9191919191919192,
+      "grad_norm": 1.3433420658111572,
+      "learning_rate": 2.7272727272727272e-06,
+      "loss": 7.3152,
+      "step": 91
+    },
+    {
+      "epoch": 0.9292929292929293,
+      "grad_norm": 1.0658912658691406,
+      "learning_rate": 2.4242424242424244e-06,
+      "loss": 7.3189,
+      "step": 92
+    },
+    {
+      "epoch": 0.9393939393939394,
+      "grad_norm": 0.9861695170402527,
+      "learning_rate": 2.121212121212121e-06,
+      "loss": 7.2579,
+      "step": 93
+    },
+    {
+      "epoch": 0.9494949494949495,
+      "grad_norm": 0.9982382655143738,
+      "learning_rate": 1.8181818181818183e-06,
+      "loss": 7.2653,
+      "step": 94
+    },
+    {
+      "epoch": 0.9595959595959596,
+      "grad_norm": 0.8333445191383362,
+      "learning_rate": 1.5151515151515152e-06,
+      "loss": 7.159,
+      "step": 95
+    },
+    {
+      "epoch": 0.9696969696969697,
+      "grad_norm": 0.7748318910598755,
+      "learning_rate": 1.2121212121212122e-06,
+      "loss": 7.1589,
+      "step": 96
+    },
+    {
+      "epoch": 0.9797979797979798,
+      "grad_norm": 0.8152617812156677,
+      "learning_rate": 9.090909090909091e-07,
+      "loss": 7.2117,
+      "step": 97
+    },
+    {
+      "epoch": 0.98989898989899,
+      "grad_norm": 1.3867417573928833,
+      "learning_rate": 6.060606060606061e-07,
+      "loss": 6.9967,
+      "step": 98
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 1.3770259618759155,
+      "learning_rate": 3.0303030303030305e-07,
+      "loss": 7.3112,
+      "step": 99
+    },
+    {
+      "epoch": 1.0,
+      "eval_audio_cosine_sim": 0.4540814161300659,
+      "eval_loss": 3.913311243057251,
+      "eval_runtime": 301.4584,
+      "eval_samples_per_second": 0.03,
+      "eval_steps_per_second": 0.01,
+      "eval_text_cosine_sim": 0.24709360301494598,
+      "step": 99
+    },
+    {
+      "epoch": 1.0,
+      "step": 99,
+      "total_flos": 2077734266890272.0,
+      "train_loss": 7.893082406785753,
+      "train_runtime": 1993.2394,
+      "train_samples_per_second": 0.793,
+      "train_steps_per_second": 0.05
+    }
+  ],
+  "logging_steps": 1.0,
+  "max_steps": 99,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "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": 2077734266890272.0,
+  "train_batch_size": 16,
+  "trial_name": null,
+  "trial_params": null
+}