bge-small-hotpotwa-matryoshka-fine-tuned-50/checkpoint-4000/trainer_state.json

{
  "best_metric": 0.8931613819214387,
  "best_model_checkpoint": "bge-small-hotpotwa-matryoshka-fine-tuned-50/checkpoint-500",
  "epoch": 26.924694993689524,
  "eval_steps": 50,
  "global_step": 4000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.33655868742111905,
      "grad_norm": 1.7359095811843872,
      "learning_rate": 1.3513513513513515e-06,
      "loss": 19.5758,
      "step": 50
    },
    {
      "epoch": 0.33655868742111905,
      "eval_dim_128_cosine_accuracy": 0.9551585423568386,
      "eval_dim_128_dot_accuracy": 0.08980123047799338,
      "eval_dim_128_euclidean_accuracy": 0.9530288689067676,
      "eval_dim_128_manhattan_accuracy": 0.9527922385234264,
      "eval_dim_128_max_accuracy": 0.9551585423568386,
      "eval_dim_256_cosine_accuracy": 0.966280170373876,
      "eval_dim_256_dot_accuracy": 0.042711784193090394,
      "eval_dim_256_euclidean_accuracy": 0.9659252247988642,
      "eval_dim_256_manhattan_accuracy": 0.9634406057737813,
      "eval_dim_256_max_accuracy": 0.966280170373876,
      "eval_dim_384_cosine_accuracy": 0.9667534311405585,
      "eval_dim_384_dot_accuracy": 0.03324656885944155,
      "eval_dim_384_euclidean_accuracy": 0.9667534311405585,
      "eval_dim_384_manhattan_accuracy": 0.9669900615238997,
      "eval_dim_384_max_accuracy": 0.9669900615238997,
      "eval_dim_64_cosine_accuracy": 0.9358731661145291,
      "eval_dim_64_dot_accuracy": 0.1320397539044013,
      "eval_dim_64_euclidean_accuracy": 0.9345716990061524,
      "eval_dim_64_manhattan_accuracy": 0.9269995267392334,
      "eval_dim_64_max_accuracy": 0.9358731661145291,
      "eval_loss": 19.393272399902344,
      "eval_runtime": 104.7788,
      "eval_samples_per_second": 80.665,
      "eval_sequential_score": 0.9358731661145291,
      "eval_steps_per_second": 2.529,
      "step": 50
    },
    {
      "epoch": 0.6731173748422381,
      "grad_norm": 1.976278305053711,
      "learning_rate": 2.702702702702703e-06,
      "loss": 19.4573,
      "step": 100
    },
    {
      "epoch": 0.6731173748422381,
      "eval_dim_128_cosine_accuracy": 0.9570515854235684,
      "eval_dim_128_dot_accuracy": 0.06625650733554188,
      "eval_dim_128_euclidean_accuracy": 0.9589446284902982,
      "eval_dim_128_manhattan_accuracy": 0.9557501183151916,
      "eval_dim_128_max_accuracy": 0.9589446284902982,
      "eval_dim_256_cosine_accuracy": 0.9646237576904875,
      "eval_dim_256_dot_accuracy": 0.04046379555134879,
      "eval_dim_256_euclidean_accuracy": 0.9650970184571699,
      "eval_dim_256_manhattan_accuracy": 0.9632039753904401,
      "eval_dim_256_max_accuracy": 0.9650970184571699,
      "eval_dim_384_cosine_accuracy": 0.9653336488405111,
      "eval_dim_384_dot_accuracy": 0.03466635115948888,
      "eval_dim_384_euclidean_accuracy": 0.9653336488405111,
      "eval_dim_384_manhattan_accuracy": 0.9646237576904875,
      "eval_dim_384_max_accuracy": 0.9653336488405111,
      "eval_dim_64_cosine_accuracy": 0.9449834358731661,
      "eval_dim_64_dot_accuracy": 0.08932796971131093,
      "eval_dim_64_euclidean_accuracy": 0.9461665877898722,
      "eval_dim_64_manhattan_accuracy": 0.9420255560814008,
      "eval_dim_64_max_accuracy": 0.9461665877898722,
      "eval_loss": 19.097097396850586,
      "eval_runtime": 103.9699,
      "eval_samples_per_second": 81.293,
      "eval_sequential_score": 0.9449834358731661,
      "eval_steps_per_second": 2.549,
      "step": 100
    },
    {
      "epoch": 1.0096760622633572,
      "grad_norm": 2.1209616661071777,
      "learning_rate": 4.0540540540540545e-06,
      "loss": 19.1409,
      "step": 150
    },
    {
      "epoch": 1.0096760622633572,
      "eval_dim_128_cosine_accuracy": 0.9384761003312825,
      "eval_dim_128_dot_accuracy": 0.06897775674396593,
      "eval_dim_128_euclidean_accuracy": 0.9421438712730714,
      "eval_dim_128_manhattan_accuracy": 0.939540937056318,
      "eval_dim_128_max_accuracy": 0.9421438712730714,
      "eval_dim_256_cosine_accuracy": 0.9434453383814482,
      "eval_dim_256_dot_accuracy": 0.05797444391859915,
      "eval_dim_256_euclidean_accuracy": 0.9436819687647894,
      "eval_dim_256_manhattan_accuracy": 0.9423805016564126,
      "eval_dim_256_max_accuracy": 0.9436819687647894,
      "eval_dim_384_cosine_accuracy": 0.9473497397065783,
      "eval_dim_384_dot_accuracy": 0.05265026029342167,
      "eval_dim_384_euclidean_accuracy": 0.9473497397065783,
      "eval_dim_384_manhattan_accuracy": 0.9458116422148604,
      "eval_dim_384_max_accuracy": 0.9473497397065783,
      "eval_dim_64_cosine_accuracy": 0.9306672976810223,
      "eval_dim_64_dot_accuracy": 0.07749645054424988,
      "eval_dim_64_euclidean_accuracy": 0.9332702318977757,
      "eval_dim_64_manhattan_accuracy": 0.9320870799810695,
      "eval_dim_64_max_accuracy": 0.9332702318977757,
      "eval_loss": 18.4069766998291,
      "eval_runtime": 103.2125,
      "eval_samples_per_second": 81.889,
      "eval_sequential_score": 0.9306672976810223,
      "eval_steps_per_second": 2.568,
      "step": 150
    },
    {
      "epoch": 1.3462347496844762,
      "grad_norm": 1.658170461654663,
      "learning_rate": 5.405405405405406e-06,
      "loss": 18.6431,
      "step": 200
    },
    {
      "epoch": 1.3462347496844762,
      "eval_dim_128_cosine_accuracy": 0.9125650733554188,
      "eval_dim_128_dot_accuracy": 0.08826313298627544,
      "eval_dim_128_euclidean_accuracy": 0.9139848556554662,
      "eval_dim_128_manhattan_accuracy": 0.9145764316138192,
      "eval_dim_128_max_accuracy": 0.9145764316138192,
      "eval_dim_256_cosine_accuracy": 0.9163511594888784,
      "eval_dim_256_dot_accuracy": 0.08613345953620445,
      "eval_dim_256_euclidean_accuracy": 0.9163511594888784,
      "eval_dim_256_manhattan_accuracy": 0.9151680075721723,
      "eval_dim_256_max_accuracy": 0.9163511594888784,
      "eval_dim_384_cosine_accuracy": 0.9183625177472787,
      "eval_dim_384_dot_accuracy": 0.08163748225272124,
      "eval_dim_384_euclidean_accuracy": 0.9183625177472787,
      "eval_dim_384_manhattan_accuracy": 0.9184808329389493,
      "eval_dim_384_max_accuracy": 0.9184808329389493,
      "eval_dim_64_cosine_accuracy": 0.9093705631803124,
      "eval_dim_64_dot_accuracy": 0.09477046852815901,
      "eval_dim_64_euclidean_accuracy": 0.9126833885470894,
      "eval_dim_64_manhattan_accuracy": 0.9113819214387128,
      "eval_dim_64_max_accuracy": 0.9126833885470894,
      "eval_loss": 17.32919692993164,
      "eval_runtime": 102.8811,
      "eval_samples_per_second": 82.153,
      "eval_sequential_score": 0.9093705631803124,
      "eval_steps_per_second": 2.576,
      "step": 200
    },
    {
      "epoch": 1.6827934371055953,
      "grad_norm": 1.5389924049377441,
      "learning_rate": 6.7567567567567575e-06,
      "loss": 18.2288,
      "step": 250
    },
    {
      "epoch": 1.6827934371055953,
      "eval_dim_128_cosine_accuracy": 0.9062943681968765,
      "eval_dim_128_dot_accuracy": 0.09311405584477046,
      "eval_dim_128_euclidean_accuracy": 0.9062943681968765,
      "eval_dim_128_manhattan_accuracy": 0.9062943681968765,
      "eval_dim_128_max_accuracy": 0.9062943681968765,
      "eval_dim_256_cosine_accuracy": 0.9071225745385707,
      "eval_dim_256_dot_accuracy": 0.09335068622811168,
      "eval_dim_256_euclidean_accuracy": 0.907950780880265,
      "eval_dim_256_manhattan_accuracy": 0.9093705631803124,
      "eval_dim_256_max_accuracy": 0.9093705631803124,
      "eval_dim_384_cosine_accuracy": 0.9099621391386654,
      "eval_dim_384_dot_accuracy": 0.0900378608613346,
      "eval_dim_384_euclidean_accuracy": 0.9099621391386654,
      "eval_dim_384_manhattan_accuracy": 0.9087789872219593,
      "eval_dim_384_max_accuracy": 0.9099621391386654,
      "eval_dim_64_cosine_accuracy": 0.9022716516800757,
      "eval_dim_64_dot_accuracy": 0.09962139138665405,
      "eval_dim_64_euclidean_accuracy": 0.9046379555134879,
      "eval_dim_64_manhattan_accuracy": 0.9040463795551349,
      "eval_dim_64_max_accuracy": 0.9046379555134879,
      "eval_loss": 16.875099182128906,
      "eval_runtime": 104.7249,
      "eval_samples_per_second": 80.707,
      "eval_sequential_score": 0.9022716516800757,
      "eval_steps_per_second": 2.53,
      "step": 250
    },
    {
      "epoch": 2.0193521245267143,
      "grad_norm": 1.4371246099472046,
      "learning_rate": 8.108108108108109e-06,
      "loss": 18.0425,
      "step": 300
    },
    {
      "epoch": 2.0193521245267143,
      "eval_dim_128_cosine_accuracy": 0.9020350212967345,
      "eval_dim_128_dot_accuracy": 0.09772834831992427,
      "eval_dim_128_euclidean_accuracy": 0.9035731187884525,
      "eval_dim_128_manhattan_accuracy": 0.9044013251301467,
      "eval_dim_128_max_accuracy": 0.9044013251301467,
      "eval_dim_256_cosine_accuracy": 0.9032181732134406,
      "eval_dim_256_dot_accuracy": 0.09690014197823,
      "eval_dim_256_euclidean_accuracy": 0.90309985802177,
      "eval_dim_256_manhattan_accuracy": 0.9042830099384761,
      "eval_dim_256_max_accuracy": 0.9042830099384761,
      "eval_dim_384_cosine_accuracy": 0.9045196403218173,
      "eval_dim_384_dot_accuracy": 0.09548035967818268,
      "eval_dim_384_euclidean_accuracy": 0.9045196403218173,
      "eval_dim_384_manhattan_accuracy": 0.9049929010884997,
      "eval_dim_384_max_accuracy": 0.9049929010884997,
      "eval_dim_64_cosine_accuracy": 0.8989588263132986,
      "eval_dim_64_dot_accuracy": 0.10234264079507809,
      "eval_dim_64_euclidean_accuracy": 0.9016800757217227,
      "eval_dim_64_manhattan_accuracy": 0.9016800757217227,
      "eval_dim_64_max_accuracy": 0.9016800757217227,
      "eval_loss": 16.69808578491211,
      "eval_runtime": 103.4615,
      "eval_samples_per_second": 81.692,
      "eval_sequential_score": 0.8989588263132986,
      "eval_steps_per_second": 2.561,
      "step": 300
    },
    {
      "epoch": 2.3559108119478336,
      "grad_norm": 1.386720895767212,
      "learning_rate": 9.45945945945946e-06,
      "loss": 17.9458,
      "step": 350
    },
    {
      "epoch": 2.3559108119478336,
      "eval_dim_128_cosine_accuracy": 0.9036914339801231,
      "eval_dim_128_dot_accuracy": 0.09761003312825367,
      "eval_dim_128_euclidean_accuracy": 0.9034548035967819,
      "eval_dim_128_manhattan_accuracy": 0.9016800757217227,
      "eval_dim_128_max_accuracy": 0.9036914339801231,
      "eval_dim_256_cosine_accuracy": 0.9013251301467108,
      "eval_dim_256_dot_accuracy": 0.09855655466161856,
      "eval_dim_256_euclidean_accuracy": 0.9015617605300521,
      "eval_dim_256_manhattan_accuracy": 0.9022716516800757,
      "eval_dim_256_max_accuracy": 0.9022716516800757,
      "eval_dim_384_cosine_accuracy": 0.9021533364884051,
      "eval_dim_384_dot_accuracy": 0.09784666351159489,
      "eval_dim_384_euclidean_accuracy": 0.9021533364884051,
      "eval_dim_384_manhattan_accuracy": 0.9039280643634643,
      "eval_dim_384_max_accuracy": 0.9039280643634643,
      "eval_dim_64_cosine_accuracy": 0.8983672503549456,
      "eval_dim_64_dot_accuracy": 0.10352579271178419,
      "eval_dim_64_euclidean_accuracy": 0.8995504022716517,
      "eval_dim_64_manhattan_accuracy": 0.8981306199716044,
      "eval_dim_64_max_accuracy": 0.8995504022716517,
      "eval_loss": 16.615509033203125,
      "eval_runtime": 103.1308,
      "eval_samples_per_second": 81.954,
      "eval_sequential_score": 0.8983672503549456,
      "eval_steps_per_second": 2.57,
      "step": 350
    },
    {
      "epoch": 2.6924694993689524,
      "grad_norm": 1.4882862567901611,
      "learning_rate": 1.0810810810810812e-05,
      "loss": 17.8525,
      "step": 400
    },
    {
      "epoch": 2.6924694993689524,
      "eval_dim_128_cosine_accuracy": 0.8977756743965926,
      "eval_dim_128_dot_accuracy": 0.10269758637008992,
      "eval_dim_128_euclidean_accuracy": 0.9006152389966872,
      "eval_dim_128_manhattan_accuracy": 0.900378608613346,
      "eval_dim_128_max_accuracy": 0.9006152389966872,
      "eval_dim_256_cosine_accuracy": 0.8970657832465688,
      "eval_dim_256_dot_accuracy": 0.10269758637008992,
      "eval_dim_256_euclidean_accuracy": 0.8980123047799338,
      "eval_dim_256_manhattan_accuracy": 0.8971840984382394,
      "eval_dim_256_max_accuracy": 0.8980123047799338,
      "eval_dim_384_cosine_accuracy": 0.8974207288215806,
      "eval_dim_384_dot_accuracy": 0.1025792711784193,
      "eval_dim_384_euclidean_accuracy": 0.8974207288215806,
      "eval_dim_384_manhattan_accuracy": 0.898248935163275,
      "eval_dim_384_max_accuracy": 0.898248935163275,
      "eval_dim_64_cosine_accuracy": 0.8948177946048272,
      "eval_dim_64_dot_accuracy": 0.10636535731187885,
      "eval_dim_64_euclidean_accuracy": 0.8969474680548982,
      "eval_dim_64_manhattan_accuracy": 0.8948177946048272,
      "eval_dim_64_max_accuracy": 0.8969474680548982,
      "eval_loss": 16.553625106811523,
      "eval_runtime": 103.3808,
      "eval_samples_per_second": 81.756,
      "eval_sequential_score": 0.8948177946048272,
      "eval_steps_per_second": 2.563,
      "step": 400
    },
    {
      "epoch": 3.0290281867900717,
      "grad_norm": 1.5986053943634033,
      "learning_rate": 1.2162162162162164e-05,
      "loss": 17.7529,
      "step": 450
    },
    {
      "epoch": 3.0290281867900717,
      "eval_dim_128_cosine_accuracy": 0.8980123047799338,
      "eval_dim_128_dot_accuracy": 0.10340747752011359,
      "eval_dim_128_euclidean_accuracy": 0.8997870326549929,
      "eval_dim_128_manhattan_accuracy": 0.8996687174633223,
      "eval_dim_128_max_accuracy": 0.8997870326549929,
      "eval_dim_256_cosine_accuracy": 0.8956460009465216,
      "eval_dim_256_dot_accuracy": 0.10399905347846664,
      "eval_dim_256_euclidean_accuracy": 0.8970657832465688,
      "eval_dim_256_manhattan_accuracy": 0.8960009465215334,
      "eval_dim_256_max_accuracy": 0.8970657832465688,
      "eval_dim_384_cosine_accuracy": 0.8952910553715097,
      "eval_dim_384_dot_accuracy": 0.1047089446284903,
      "eval_dim_384_euclidean_accuracy": 0.8952910553715097,
      "eval_dim_384_manhattan_accuracy": 0.8971840984382394,
      "eval_dim_384_max_accuracy": 0.8971840984382394,
      "eval_dim_64_cosine_accuracy": 0.8950544249881685,
      "eval_dim_64_dot_accuracy": 0.10541883577851396,
      "eval_dim_64_euclidean_accuracy": 0.8969474680548982,
      "eval_dim_64_manhattan_accuracy": 0.8948177946048272,
      "eval_dim_64_max_accuracy": 0.8969474680548982,
      "eval_loss": 16.51355743408203,
      "eval_runtime": 104.654,
      "eval_samples_per_second": 80.761,
      "eval_sequential_score": 0.8950544249881685,
      "eval_steps_per_second": 2.532,
      "step": 450
    },
    {
      "epoch": 3.3655868742111905,
      "grad_norm": 1.8756661415100098,
      "learning_rate": 1.3513513513513515e-05,
      "loss": 17.6709,
      "step": 500
    },
    {
      "epoch": 3.3655868742111905,
      "eval_dim_128_cosine_accuracy": 0.8931613819214387,
      "eval_dim_128_dot_accuracy": 0.10766682442025556,
      "eval_dim_128_euclidean_accuracy": 0.8944628490298154,
      "eval_dim_128_manhattan_accuracy": 0.8942262186464742,
      "eval_dim_128_max_accuracy": 0.8944628490298154,
      "eval_dim_256_cosine_accuracy": 0.8913866540463795,
      "eval_dim_256_dot_accuracy": 0.10896829152863227,
      "eval_dim_256_euclidean_accuracy": 0.8937529578797918,
      "eval_dim_256_manhattan_accuracy": 0.8937529578797918,
      "eval_dim_256_max_accuracy": 0.8937529578797918,
      "eval_dim_384_cosine_accuracy": 0.8928064363464269,
      "eval_dim_384_dot_accuracy": 0.10719356365357312,
      "eval_dim_384_euclidean_accuracy": 0.8928064363464269,
      "eval_dim_384_manhattan_accuracy": 0.8932796971131093,
      "eval_dim_384_max_accuracy": 0.8932796971131093,
      "eval_dim_64_cosine_accuracy": 0.8906767628963559,
      "eval_dim_64_dot_accuracy": 0.11121628017037388,
      "eval_dim_64_euclidean_accuracy": 0.8911500236630383,
      "eval_dim_64_manhattan_accuracy": 0.8893752957879791,
      "eval_dim_64_max_accuracy": 0.8911500236630383,
      "eval_loss": 16.4824161529541,
      "eval_runtime": 103.2754,
      "eval_samples_per_second": 81.839,
      "eval_sequential_score": 0.8906767628963559,
      "eval_steps_per_second": 2.566,
      "step": 500
    },
    {
      "epoch": 3.70214556163231,
      "grad_norm": 2.3590304851531982,
      "learning_rate": 1.4864864864864865e-05,
      "loss": 17.5348,
      "step": 550
    },
    {
      "epoch": 3.70214556163231,
      "eval_dim_128_cosine_accuracy": 0.8862991008045433,
      "eval_dim_128_dot_accuracy": 0.11500236630383341,
      "eval_dim_128_euclidean_accuracy": 0.8864174159962139,
      "eval_dim_128_manhattan_accuracy": 0.8858258400378609,
      "eval_dim_128_max_accuracy": 0.8864174159962139,
      "eval_dim_256_cosine_accuracy": 0.8858258400378609,
      "eval_dim_256_dot_accuracy": 0.11358258400378608,
      "eval_dim_256_euclidean_accuracy": 0.8867723615712257,
      "eval_dim_256_manhattan_accuracy": 0.8858258400378609,
      "eval_dim_256_max_accuracy": 0.8867723615712257,
      "eval_dim_384_cosine_accuracy": 0.8859441552295315,
      "eval_dim_384_dot_accuracy": 0.11405584477046853,
      "eval_dim_384_euclidean_accuracy": 0.8859441552295315,
      "eval_dim_384_manhattan_accuracy": 0.88760056791292,
      "eval_dim_384_max_accuracy": 0.88760056791292,
      "eval_dim_64_cosine_accuracy": 0.884879318504496,
      "eval_dim_64_dot_accuracy": 0.11985328916232844,
      "eval_dim_64_euclidean_accuracy": 0.8845243729294842,
      "eval_dim_64_manhattan_accuracy": 0.8828679602460956,
      "eval_dim_64_max_accuracy": 0.884879318504496,
      "eval_loss": 16.463218688964844,
      "eval_runtime": 103.2788,
      "eval_samples_per_second": 81.837,
      "eval_sequential_score": 0.884879318504496,
      "eval_steps_per_second": 2.566,
      "step": 550
    },
    {
      "epoch": 4.038704249053429,
      "grad_norm": 2.6120336055755615,
      "learning_rate": 1.6216216216216218e-05,
      "loss": 17.4198,
      "step": 600
    },
    {
      "epoch": 4.038704249053429,
      "eval_dim_128_cosine_accuracy": 0.8852342640795078,
      "eval_dim_128_dot_accuracy": 0.11748698532891623,
      "eval_dim_128_euclidean_accuracy": 0.8846426881211548,
      "eval_dim_128_manhattan_accuracy": 0.8859441552295315,
      "eval_dim_128_max_accuracy": 0.8859441552295315,
      "eval_dim_256_cosine_accuracy": 0.8861807856128727,
      "eval_dim_256_dot_accuracy": 0.1137008991954567,
      "eval_dim_256_euclidean_accuracy": 0.8871273071462376,
      "eval_dim_256_manhattan_accuracy": 0.8866540463795551,
      "eval_dim_256_max_accuracy": 0.8871273071462376,
      "eval_dim_384_cosine_accuracy": 0.8859441552295315,
      "eval_dim_384_dot_accuracy": 0.11405584477046853,
      "eval_dim_384_euclidean_accuracy": 0.8859441552295315,
      "eval_dim_384_manhattan_accuracy": 0.8847610033128254,
      "eval_dim_384_max_accuracy": 0.8859441552295315,
      "eval_dim_64_cosine_accuracy": 0.8839327969711311,
      "eval_dim_64_dot_accuracy": 0.12103644107903455,
      "eval_dim_64_euclidean_accuracy": 0.8861807856128727,
      "eval_dim_64_manhattan_accuracy": 0.8857075248461902,
      "eval_dim_64_max_accuracy": 0.8861807856128727,
      "eval_loss": 16.46009063720703,
      "eval_runtime": 104.1113,
      "eval_samples_per_second": 81.182,
      "eval_sequential_score": 0.8839327969711311,
      "eval_steps_per_second": 2.545,
      "step": 600
    },
    {
      "epoch": 4.375262936474548,
      "grad_norm": 2.63383412361145,
      "learning_rate": 1.756756756756757e-05,
      "loss": 17.3673,
      "step": 650
    },
    {
      "epoch": 4.375262936474548,
      "eval_dim_128_cosine_accuracy": 0.8853525792711784,
      "eval_dim_128_dot_accuracy": 0.1160672030288689,
      "eval_dim_128_euclidean_accuracy": 0.8867723615712257,
      "eval_dim_128_manhattan_accuracy": 0.8855892096545196,
      "eval_dim_128_max_accuracy": 0.8867723615712257,
      "eval_dim_256_cosine_accuracy": 0.8864174159962139,
      "eval_dim_256_dot_accuracy": 0.11417415996213914,
      "eval_dim_256_euclidean_accuracy": 0.8871273071462376,
      "eval_dim_256_manhattan_accuracy": 0.8862991008045433,
      "eval_dim_256_max_accuracy": 0.8871273071462376,
      "eval_dim_384_cosine_accuracy": 0.8865357311878845,
      "eval_dim_384_dot_accuracy": 0.11346426881211548,
      "eval_dim_384_euclidean_accuracy": 0.8865357311878845,
      "eval_dim_384_manhattan_accuracy": 0.8861807856128727,
      "eval_dim_384_max_accuracy": 0.8865357311878845,
      "eval_dim_64_cosine_accuracy": 0.8841694273544723,
      "eval_dim_64_dot_accuracy": 0.12091812588736393,
      "eval_dim_64_euclidean_accuracy": 0.883341221012778,
      "eval_dim_64_manhattan_accuracy": 0.8828679602460956,
      "eval_dim_64_max_accuracy": 0.8841694273544723,
      "eval_loss": 16.440513610839844,
      "eval_runtime": 102.5958,
      "eval_samples_per_second": 82.382,
      "eval_sequential_score": 0.8841694273544723,
      "eval_steps_per_second": 2.583,
      "step": 650
    },
    {
      "epoch": 4.711821623895667,
      "grad_norm": 3.044569730758667,
      "learning_rate": 1.891891891891892e-05,
      "loss": 17.2603,
      "step": 700
    },
    {
      "epoch": 4.711821623895667,
      "eval_dim_128_cosine_accuracy": 0.8834595362044486,
      "eval_dim_128_dot_accuracy": 0.11772361571225745,
      "eval_dim_128_euclidean_accuracy": 0.8835778513961192,
      "eval_dim_128_manhattan_accuracy": 0.8840511121628017,
      "eval_dim_128_max_accuracy": 0.8840511121628017,
      "eval_dim_256_cosine_accuracy": 0.8838144817794605,
      "eval_dim_256_dot_accuracy": 0.11571225745385708,
      "eval_dim_256_euclidean_accuracy": 0.8838144817794605,
      "eval_dim_256_manhattan_accuracy": 0.8839327969711311,
      "eval_dim_256_max_accuracy": 0.8839327969711311,
      "eval_dim_384_cosine_accuracy": 0.8838144817794605,
      "eval_dim_384_dot_accuracy": 0.11618551822053952,
      "eval_dim_384_euclidean_accuracy": 0.8838144817794605,
      "eval_dim_384_manhattan_accuracy": 0.8847610033128254,
      "eval_dim_384_max_accuracy": 0.8847610033128254,
      "eval_dim_64_cosine_accuracy": 0.8807382867960246,
      "eval_dim_64_dot_accuracy": 0.12328442972077615,
      "eval_dim_64_euclidean_accuracy": 0.8814481779460482,
      "eval_dim_64_manhattan_accuracy": 0.8810932323710364,
      "eval_dim_64_max_accuracy": 0.8814481779460482,
      "eval_loss": 16.435609817504883,
      "eval_runtime": 103.6437,
      "eval_samples_per_second": 81.549,
      "eval_sequential_score": 0.8807382867960246,
      "eval_steps_per_second": 2.557,
      "step": 700
    },
    {
      "epoch": 5.0483803113167856,
      "grad_norm": 3.3264880180358887,
      "learning_rate": 1.9999888744757143e-05,
      "loss": 17.1807,
      "step": 750
    },
    {
      "epoch": 5.0483803113167856,
      "eval_dim_128_cosine_accuracy": 0.8849976336961666,
      "eval_dim_128_dot_accuracy": 0.11654046379555134,
      "eval_dim_128_euclidean_accuracy": 0.884879318504496,
      "eval_dim_128_manhattan_accuracy": 0.8838144817794605,
      "eval_dim_128_max_accuracy": 0.8849976336961666,
      "eval_dim_256_cosine_accuracy": 0.8864174159962139,
      "eval_dim_256_dot_accuracy": 0.11417415996213914,
      "eval_dim_256_euclidean_accuracy": 0.8852342640795078,
      "eval_dim_256_manhattan_accuracy": 0.8857075248461902,
      "eval_dim_256_max_accuracy": 0.8864174159962139,
      "eval_dim_384_cosine_accuracy": 0.8859441552295315,
      "eval_dim_384_dot_accuracy": 0.11405584477046853,
      "eval_dim_384_euclidean_accuracy": 0.8859441552295315,
      "eval_dim_384_manhattan_accuracy": 0.8855892096545196,
      "eval_dim_384_max_accuracy": 0.8859441552295315,
      "eval_dim_64_cosine_accuracy": 0.8838144817794605,
      "eval_dim_64_dot_accuracy": 0.12079981069569333,
      "eval_dim_64_euclidean_accuracy": 0.8844060577378136,
      "eval_dim_64_manhattan_accuracy": 0.8834595362044486,
      "eval_dim_64_max_accuracy": 0.8844060577378136,
      "eval_loss": 16.444347381591797,
      "eval_runtime": 103.5226,
      "eval_samples_per_second": 81.644,
      "eval_sequential_score": 0.8838144817794605,
      "eval_steps_per_second": 2.56,
      "step": 750
    },
    {
      "epoch": 5.384938998737905,
      "grad_norm": 2.7032034397125244,
      "learning_rate": 1.999599507118322e-05,
      "loss": 17.1629,
      "step": 800
    },
    {
      "epoch": 5.384938998737905,
      "eval_dim_128_cosine_accuracy": 0.8847610033128254,
      "eval_dim_128_dot_accuracy": 0.11701372456223379,
      "eval_dim_128_euclidean_accuracy": 0.8859441552295315,
      "eval_dim_128_manhattan_accuracy": 0.884879318504496,
      "eval_dim_128_max_accuracy": 0.8859441552295315,
      "eval_dim_256_cosine_accuracy": 0.8861807856128727,
      "eval_dim_256_dot_accuracy": 0.11417415996213914,
      "eval_dim_256_euclidean_accuracy": 0.8859441552295315,
      "eval_dim_256_manhattan_accuracy": 0.8853525792711784,
      "eval_dim_256_max_accuracy": 0.8861807856128727,
      "eval_dim_384_cosine_accuracy": 0.8866540463795551,
      "eval_dim_384_dot_accuracy": 0.11334595362044486,
      "eval_dim_384_euclidean_accuracy": 0.8866540463795551,
      "eval_dim_384_manhattan_accuracy": 0.8862991008045433,
      "eval_dim_384_max_accuracy": 0.8866540463795551,
      "eval_dim_64_cosine_accuracy": 0.8841694273544723,
      "eval_dim_64_dot_accuracy": 0.11831519167061051,
      "eval_dim_64_euclidean_accuracy": 0.8841694273544723,
      "eval_dim_64_manhattan_accuracy": 0.8839327969711311,
      "eval_dim_64_max_accuracy": 0.8841694273544723,
      "eval_loss": 16.420166015625,
      "eval_runtime": 103.5297,
      "eval_samples_per_second": 81.638,
      "eval_sequential_score": 0.8841694273544723,
      "eval_steps_per_second": 2.56,
      "step": 800
    },
    {
      "epoch": 5.721497686159024,
      "grad_norm": 3.8163998126983643,
      "learning_rate": 1.9986541110764565e-05,
      "loss": 17.0747,
      "step": 850
    },
    {
      "epoch": 5.721497686159024,
      "eval_dim_128_cosine_accuracy": 0.8853525792711784,
      "eval_dim_128_dot_accuracy": 0.11618551822053952,
      "eval_dim_128_euclidean_accuracy": 0.8835778513961192,
      "eval_dim_128_manhattan_accuracy": 0.8845243729294842,
      "eval_dim_128_max_accuracy": 0.8853525792711784,
      "eval_dim_256_cosine_accuracy": 0.8874822527212494,
      "eval_dim_256_dot_accuracy": 0.11358258400378608,
      "eval_dim_256_euclidean_accuracy": 0.8864174159962139,
      "eval_dim_256_manhattan_accuracy": 0.8862991008045433,
      "eval_dim_256_max_accuracy": 0.8874822527212494,
      "eval_dim_384_cosine_accuracy": 0.8868906767628963,
      "eval_dim_384_dot_accuracy": 0.11310932323710364,
      "eval_dim_384_euclidean_accuracy": 0.8868906767628963,
      "eval_dim_384_manhattan_accuracy": 0.8862991008045433,
      "eval_dim_384_max_accuracy": 0.8868906767628963,
      "eval_dim_64_cosine_accuracy": 0.8836961665877898,
      "eval_dim_64_dot_accuracy": 0.11867013724562234,
      "eval_dim_64_euclidean_accuracy": 0.8831045906294368,
      "eval_dim_64_manhattan_accuracy": 0.8832229058211074,
      "eval_dim_64_max_accuracy": 0.8836961665877898,
      "eval_loss": 16.416208267211914,
      "eval_runtime": 103.4694,
      "eval_samples_per_second": 81.686,
      "eval_sequential_score": 0.8836961665877898,
      "eval_steps_per_second": 2.561,
      "step": 850
    },
    {
      "epoch": 6.058056373580143,
      "grad_norm": 3.9848620891571045,
      "learning_rate": 1.9971532122280466e-05,
      "loss": 17.0161,
      "step": 900
    },
    {
      "epoch": 6.058056373580143,
      "eval_dim_128_cosine_accuracy": 0.8852342640795078,
      "eval_dim_128_dot_accuracy": 0.11618551822053952,
      "eval_dim_128_euclidean_accuracy": 0.8852342640795078,
      "eval_dim_128_manhattan_accuracy": 0.8846426881211548,
      "eval_dim_128_max_accuracy": 0.8852342640795078,
      "eval_dim_256_cosine_accuracy": 0.8862991008045433,
      "eval_dim_256_dot_accuracy": 0.11417415996213914,
      "eval_dim_256_euclidean_accuracy": 0.8858258400378609,
      "eval_dim_256_manhattan_accuracy": 0.8853525792711784,
      "eval_dim_256_max_accuracy": 0.8862991008045433,
      "eval_dim_384_cosine_accuracy": 0.8855892096545196,
      "eval_dim_384_dot_accuracy": 0.11441079034548036,
      "eval_dim_384_euclidean_accuracy": 0.8855892096545196,
      "eval_dim_384_manhattan_accuracy": 0.885470894462849,
      "eval_dim_384_max_accuracy": 0.8855892096545196,
      "eval_dim_64_cosine_accuracy": 0.8855892096545196,
      "eval_dim_64_dot_accuracy": 0.11831519167061051,
      "eval_dim_64_euclidean_accuracy": 0.885470894462849,
      "eval_dim_64_manhattan_accuracy": 0.8834595362044486,
      "eval_dim_64_max_accuracy": 0.8855892096545196,
      "eval_loss": 16.419212341308594,
      "eval_runtime": 104.3001,
      "eval_samples_per_second": 81.035,
      "eval_sequential_score": 0.8855892096545196,
      "eval_steps_per_second": 2.541,
      "step": 900
    },
    {
      "epoch": 6.394615061001262,
      "grad_norm": 4.083323001861572,
      "learning_rate": 1.995097645450266e-05,
      "loss": 17.0146,
      "step": 950
    },
    {
      "epoch": 6.394615061001262,
      "eval_dim_128_cosine_accuracy": 0.884879318504496,
      "eval_dim_128_dot_accuracy": 0.1171320397539044,
      "eval_dim_128_euclidean_accuracy": 0.8861807856128727,
      "eval_dim_128_manhattan_accuracy": 0.8853525792711784,
      "eval_dim_128_max_accuracy": 0.8861807856128727,
      "eval_dim_256_cosine_accuracy": 0.8853525792711784,
      "eval_dim_256_dot_accuracy": 0.11464742072882159,
      "eval_dim_256_euclidean_accuracy": 0.885470894462849,
      "eval_dim_256_manhattan_accuracy": 0.8858258400378609,
      "eval_dim_256_max_accuracy": 0.8858258400378609,
      "eval_dim_384_cosine_accuracy": 0.8855892096545196,
      "eval_dim_384_dot_accuracy": 0.11441079034548036,
      "eval_dim_384_euclidean_accuracy": 0.8855892096545196,
      "eval_dim_384_manhattan_accuracy": 0.8864174159962139,
      "eval_dim_384_max_accuracy": 0.8864174159962139,
      "eval_dim_64_cosine_accuracy": 0.8844060577378136,
      "eval_dim_64_dot_accuracy": 0.11796024609559867,
      "eval_dim_64_euclidean_accuracy": 0.8852342640795078,
      "eval_dim_64_manhattan_accuracy": 0.8844060577378136,
      "eval_dim_64_max_accuracy": 0.8852342640795078,
      "eval_loss": 16.403297424316406,
      "eval_runtime": 102.2875,
      "eval_samples_per_second": 82.63,
      "eval_sequential_score": 0.8844060577378136,
      "eval_steps_per_second": 2.591,
      "step": 950
    },
    {
      "epoch": 6.731173748422381,
      "grad_norm": 3.874021291732788,
      "learning_rate": 1.992488554155135e-05,
      "loss": 16.9393,
      "step": 1000
    },
    {
      "epoch": 6.731173748422381,
      "eval_dim_128_cosine_accuracy": 0.8828679602460956,
      "eval_dim_128_dot_accuracy": 0.11784193090392807,
      "eval_dim_128_euclidean_accuracy": 0.8846426881211548,
      "eval_dim_128_manhattan_accuracy": 0.8841694273544723,
      "eval_dim_128_max_accuracy": 0.8846426881211548,
      "eval_dim_256_cosine_accuracy": 0.8839327969711311,
      "eval_dim_256_dot_accuracy": 0.1171320397539044,
      "eval_dim_256_euclidean_accuracy": 0.8840511121628017,
      "eval_dim_256_manhattan_accuracy": 0.8852342640795078,
      "eval_dim_256_max_accuracy": 0.8852342640795078,
      "eval_dim_384_cosine_accuracy": 0.8847610033128254,
      "eval_dim_384_dot_accuracy": 0.11523899668717463,
      "eval_dim_384_euclidean_accuracy": 0.8847610033128254,
      "eval_dim_384_manhattan_accuracy": 0.8852342640795078,
      "eval_dim_384_max_accuracy": 0.8852342640795078,
      "eval_dim_64_cosine_accuracy": 0.8834595362044486,
      "eval_dim_64_dot_accuracy": 0.11831519167061051,
      "eval_dim_64_euclidean_accuracy": 0.8835778513961192,
      "eval_dim_64_manhattan_accuracy": 0.8820397539044014,
      "eval_dim_64_max_accuracy": 0.8835778513961192,
      "eval_loss": 16.40532684326172,
      "eval_runtime": 104.0121,
      "eval_samples_per_second": 81.26,
      "eval_sequential_score": 0.8834595362044486,
      "eval_steps_per_second": 2.548,
      "step": 1000
    },
    {
      "epoch": 7.0677324358435,
      "grad_norm": 4.689154148101807,
      "learning_rate": 1.9893273896534936e-05,
      "loss": 16.899,
      "step": 1050
    },
    {
      "epoch": 7.0677324358435,
      "eval_dim_128_cosine_accuracy": 0.8826313298627544,
      "eval_dim_128_dot_accuracy": 0.11867013724562234,
      "eval_dim_128_euclidean_accuracy": 0.8823946994794132,
      "eval_dim_128_manhattan_accuracy": 0.882158069096072,
      "eval_dim_128_max_accuracy": 0.8826313298627544,
      "eval_dim_256_cosine_accuracy": 0.8828679602460956,
      "eval_dim_256_dot_accuracy": 0.11725035494557501,
      "eval_dim_256_euclidean_accuracy": 0.8831045906294368,
      "eval_dim_256_manhattan_accuracy": 0.8834595362044486,
      "eval_dim_256_max_accuracy": 0.8834595362044486,
      "eval_dim_384_cosine_accuracy": 0.883341221012778,
      "eval_dim_384_dot_accuracy": 0.11665877898722196,
      "eval_dim_384_euclidean_accuracy": 0.883341221012778,
      "eval_dim_384_manhattan_accuracy": 0.8839327969711311,
      "eval_dim_384_max_accuracy": 0.8839327969711311,
      "eval_dim_64_cosine_accuracy": 0.88180312352106,
      "eval_dim_64_dot_accuracy": 0.11890676762896356,
      "eval_dim_64_euclidean_accuracy": 0.8828679602460956,
      "eval_dim_64_manhattan_accuracy": 0.8820397539044014,
      "eval_dim_64_max_accuracy": 0.8828679602460956,
      "eval_loss": 16.416202545166016,
      "eval_runtime": 104.6249,
      "eval_samples_per_second": 80.784,
      "eval_sequential_score": 0.88180312352106,
      "eval_steps_per_second": 2.533,
      "step": 1050
    },
    {
      "epoch": 7.40429112326462,
      "grad_norm": 3.6406683921813965,
      "learning_rate": 1.9856159103477085e-05,
      "loss": 16.9112,
      "step": 1100
    },
    {
      "epoch": 7.40429112326462,
      "eval_dim_128_cosine_accuracy": 0.8828679602460956,
      "eval_dim_128_dot_accuracy": 0.11878845243729295,
      "eval_dim_128_euclidean_accuracy": 0.8828679602460956,
      "eval_dim_128_manhattan_accuracy": 0.8828679602460956,
      "eval_dim_128_max_accuracy": 0.8828679602460956,
      "eval_dim_256_cosine_accuracy": 0.8834595362044486,
      "eval_dim_256_dot_accuracy": 0.11618551822053952,
      "eval_dim_256_euclidean_accuracy": 0.8826313298627544,
      "eval_dim_256_manhattan_accuracy": 0.8840511121628017,
      "eval_dim_256_max_accuracy": 0.8840511121628017,
      "eval_dim_384_cosine_accuracy": 0.883341221012778,
      "eval_dim_384_dot_accuracy": 0.11665877898722196,
      "eval_dim_384_euclidean_accuracy": 0.883341221012778,
      "eval_dim_384_manhattan_accuracy": 0.884287742546143,
      "eval_dim_384_max_accuracy": 0.884287742546143,
      "eval_dim_64_cosine_accuracy": 0.8820397539044014,
      "eval_dim_64_dot_accuracy": 0.11914339801230478,
      "eval_dim_64_euclidean_accuracy": 0.8831045906294368,
      "eval_dim_64_manhattan_accuracy": 0.8826313298627544,
      "eval_dim_64_max_accuracy": 0.8831045906294368,
      "eval_loss": 16.405092239379883,
      "eval_runtime": 101.4605,
      "eval_samples_per_second": 83.303,
      "eval_sequential_score": 0.8820397539044014,
      "eval_steps_per_second": 2.612,
      "step": 1100
    },
    {
      "epoch": 7.740849810685738,
      "grad_norm": 4.141761302947998,
      "learning_rate": 1.9813561807535597e-05,
      "loss": 16.8508,
      "step": 1150
    },
    {
      "epoch": 7.740849810685738,
      "eval_dim_128_cosine_accuracy": 0.882158069096072,
      "eval_dim_128_dot_accuracy": 0.11878845243729295,
      "eval_dim_128_euclidean_accuracy": 0.8825130146710838,
      "eval_dim_128_manhattan_accuracy": 0.8838144817794605,
      "eval_dim_128_max_accuracy": 0.8838144817794605,
      "eval_dim_256_cosine_accuracy": 0.8825130146710838,
      "eval_dim_256_dot_accuracy": 0.11748698532891623,
      "eval_dim_256_euclidean_accuracy": 0.8831045906294368,
      "eval_dim_256_manhattan_accuracy": 0.8835778513961192,
      "eval_dim_256_max_accuracy": 0.8835778513961192,
      "eval_dim_384_cosine_accuracy": 0.8829862754377662,
      "eval_dim_384_dot_accuracy": 0.11701372456223379,
      "eval_dim_384_euclidean_accuracy": 0.8829862754377662,
      "eval_dim_384_manhattan_accuracy": 0.883341221012778,
      "eval_dim_384_max_accuracy": 0.883341221012778,
      "eval_dim_64_cosine_accuracy": 0.8820397539044014,
      "eval_dim_64_dot_accuracy": 0.12115475627070516,
      "eval_dim_64_euclidean_accuracy": 0.882158069096072,
      "eval_dim_64_manhattan_accuracy": 0.882749645054425,
      "eval_dim_64_max_accuracy": 0.882749645054425,
      "eval_loss": 16.40436363220215,
      "eval_runtime": 102.9818,
      "eval_samples_per_second": 82.073,
      "eval_sequential_score": 0.8820397539044014,
      "eval_steps_per_second": 2.573,
      "step": 1150
    },
    {
      "epoch": 8.077408498106857,
      "grad_norm": 3.7137351036071777,
      "learning_rate": 1.9765505703518494e-05,
      "loss": 16.8104,
      "step": 1200
    },
    {
      "epoch": 8.077408498106857,
      "eval_dim_128_cosine_accuracy": 0.8815664931377188,
      "eval_dim_128_dot_accuracy": 0.119380028395646,
      "eval_dim_128_euclidean_accuracy": 0.8813298627543776,
      "eval_dim_128_manhattan_accuracy": 0.8820397539044014,
      "eval_dim_128_max_accuracy": 0.8820397539044014,
      "eval_dim_256_cosine_accuracy": 0.8815664931377188,
      "eval_dim_256_dot_accuracy": 0.11796024609559867,
      "eval_dim_256_euclidean_accuracy": 0.8808566019876952,
      "eval_dim_256_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_256_max_accuracy": 0.8815664931377188,
      "eval_dim_384_cosine_accuracy": 0.8814481779460482,
      "eval_dim_384_dot_accuracy": 0.11855182205395173,
      "eval_dim_384_euclidean_accuracy": 0.8814481779460482,
      "eval_dim_384_manhattan_accuracy": 0.880619971604354,
      "eval_dim_384_max_accuracy": 0.8814481779460482,
      "eval_dim_64_cosine_accuracy": 0.8816848083293894,
      "eval_dim_64_dot_accuracy": 0.12174633222905822,
      "eval_dim_64_euclidean_accuracy": 0.880619971604354,
      "eval_dim_64_manhattan_accuracy": 0.8809749171793658,
      "eval_dim_64_max_accuracy": 0.8816848083293894,
      "eval_loss": 16.40627670288086,
      "eval_runtime": 104.9051,
      "eval_samples_per_second": 80.568,
      "eval_sequential_score": 0.8816848083293894,
      "eval_steps_per_second": 2.526,
      "step": 1200
    },
    {
      "epoch": 8.413967185527977,
      "grad_norm": 3.3535964488983154,
      "learning_rate": 1.9712017522703764e-05,
      "loss": 16.8212,
      "step": 1250
    },
    {
      "epoch": 8.413967185527977,
      "eval_dim_128_cosine_accuracy": 0.8834595362044486,
      "eval_dim_128_dot_accuracy": 0.11796024609559867,
      "eval_dim_128_euclidean_accuracy": 0.882749645054425,
      "eval_dim_128_manhattan_accuracy": 0.8825130146710838,
      "eval_dim_128_max_accuracy": 0.8834595362044486,
      "eval_dim_256_cosine_accuracy": 0.882158069096072,
      "eval_dim_256_dot_accuracy": 0.11748698532891623,
      "eval_dim_256_euclidean_accuracy": 0.8823946994794132,
      "eval_dim_256_manhattan_accuracy": 0.8819214387127308,
      "eval_dim_256_max_accuracy": 0.8823946994794132,
      "eval_dim_384_cosine_accuracy": 0.882158069096072,
      "eval_dim_384_dot_accuracy": 0.11784193090392807,
      "eval_dim_384_euclidean_accuracy": 0.882158069096072,
      "eval_dim_384_manhattan_accuracy": 0.882749645054425,
      "eval_dim_384_max_accuracy": 0.882749645054425,
      "eval_dim_64_cosine_accuracy": 0.8820397539044014,
      "eval_dim_64_dot_accuracy": 0.12091812588736393,
      "eval_dim_64_euclidean_accuracy": 0.8819214387127308,
      "eval_dim_64_manhattan_accuracy": 0.8815664931377188,
      "eval_dim_64_max_accuracy": 0.8820397539044014,
      "eval_loss": 16.40399169921875,
      "eval_runtime": 103.0829,
      "eval_samples_per_second": 81.992,
      "eval_sequential_score": 0.8820397539044014,
      "eval_steps_per_second": 2.571,
      "step": 1250
    },
    {
      "epoch": 8.750525872949096,
      "grad_norm": 4.203086853027344,
      "learning_rate": 1.9653127017970035e-05,
      "loss": 16.7743,
      "step": 1300
    },
    {
      "epoch": 8.750525872949096,
      "eval_dim_128_cosine_accuracy": 0.882158069096072,
      "eval_dim_128_dot_accuracy": 0.12020823473734027,
      "eval_dim_128_euclidean_accuracy": 0.8815664931377188,
      "eval_dim_128_manhattan_accuracy": 0.8814481779460482,
      "eval_dim_128_max_accuracy": 0.882158069096072,
      "eval_dim_256_cosine_accuracy": 0.8823946994794132,
      "eval_dim_256_dot_accuracy": 0.11878845243729295,
      "eval_dim_256_euclidean_accuracy": 0.8819214387127308,
      "eval_dim_256_manhattan_accuracy": 0.8816848083293894,
      "eval_dim_256_max_accuracy": 0.8823946994794132,
      "eval_dim_384_cosine_accuracy": 0.8816848083293894,
      "eval_dim_384_dot_accuracy": 0.11831519167061051,
      "eval_dim_384_euclidean_accuracy": 0.8816848083293894,
      "eval_dim_384_manhattan_accuracy": 0.882158069096072,
      "eval_dim_384_max_accuracy": 0.882158069096072,
      "eval_dim_64_cosine_accuracy": 0.8809749171793658,
      "eval_dim_64_dot_accuracy": 0.121509701845717,
      "eval_dim_64_euclidean_accuracy": 0.8807382867960246,
      "eval_dim_64_manhattan_accuracy": 0.881211547562707,
      "eval_dim_64_max_accuracy": 0.881211547562707,
      "eval_loss": 16.39342498779297,
      "eval_runtime": 102.6649,
      "eval_samples_per_second": 82.326,
      "eval_sequential_score": 0.8809749171793658,
      "eval_steps_per_second": 2.581,
      "step": 1300
    },
    {
      "epoch": 9.087084560370215,
      "grad_norm": 3.313908576965332,
      "learning_rate": 1.9588866947246498e-05,
      "loss": 16.7383,
      "step": 1350
    },
    {
      "epoch": 9.087084560370215,
      "eval_dim_128_cosine_accuracy": 0.8809749171793658,
      "eval_dim_128_dot_accuracy": 0.12068149550402271,
      "eval_dim_128_euclidean_accuracy": 0.8808566019876952,
      "eval_dim_128_manhattan_accuracy": 0.8814481779460482,
      "eval_dim_128_max_accuracy": 0.8814481779460482,
      "eval_dim_256_cosine_accuracy": 0.8820397539044014,
      "eval_dim_256_dot_accuracy": 0.11831519167061051,
      "eval_dim_256_euclidean_accuracy": 0.8810932323710364,
      "eval_dim_256_manhattan_accuracy": 0.881211547562707,
      "eval_dim_256_max_accuracy": 0.8820397539044014,
      "eval_dim_384_cosine_accuracy": 0.8807382867960246,
      "eval_dim_384_dot_accuracy": 0.11926171320397538,
      "eval_dim_384_euclidean_accuracy": 0.8807382867960246,
      "eval_dim_384_manhattan_accuracy": 0.8803833412210128,
      "eval_dim_384_max_accuracy": 0.8807382867960246,
      "eval_dim_64_cosine_accuracy": 0.880028395646001,
      "eval_dim_64_dot_accuracy": 0.12245622337908188,
      "eval_dim_64_euclidean_accuracy": 0.8807382867960246,
      "eval_dim_64_manhattan_accuracy": 0.8816848083293894,
      "eval_dim_64_max_accuracy": 0.8816848083293894,
      "eval_loss": 16.39626121520996,
      "eval_runtime": 105.1167,
      "eval_samples_per_second": 80.406,
      "eval_sequential_score": 0.880028395646001,
      "eval_steps_per_second": 2.521,
      "step": 1350
    },
    {
      "epoch": 9.423643247791334,
      "grad_norm": 6.617325305938721,
      "learning_rate": 1.9519273055291266e-05,
      "loss": 16.743,
      "step": 1400
    },
    {
      "epoch": 9.423643247791334,
      "eval_dim_128_cosine_accuracy": 0.8819214387127308,
      "eval_dim_128_dot_accuracy": 0.119380028395646,
      "eval_dim_128_euclidean_accuracy": 0.8826313298627544,
      "eval_dim_128_manhattan_accuracy": 0.8815664931377188,
      "eval_dim_128_max_accuracy": 0.8826313298627544,
      "eval_dim_256_cosine_accuracy": 0.882158069096072,
      "eval_dim_256_dot_accuracy": 0.11784193090392807,
      "eval_dim_256_euclidean_accuracy": 0.882158069096072,
      "eval_dim_256_manhattan_accuracy": 0.8816848083293894,
      "eval_dim_256_max_accuracy": 0.882158069096072,
      "eval_dim_384_cosine_accuracy": 0.8819214387127308,
      "eval_dim_384_dot_accuracy": 0.11807856128726929,
      "eval_dim_384_euclidean_accuracy": 0.8819214387127308,
      "eval_dim_384_manhattan_accuracy": 0.8826313298627544,
      "eval_dim_384_max_accuracy": 0.8826313298627544,
      "eval_dim_64_cosine_accuracy": 0.8797917652626597,
      "eval_dim_64_dot_accuracy": 0.12091812588736393,
      "eval_dim_64_euclidean_accuracy": 0.8810932323710364,
      "eval_dim_64_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_64_max_accuracy": 0.8810932323710364,
      "eval_loss": 16.406700134277344,
      "eval_runtime": 101.1577,
      "eval_samples_per_second": 83.553,
      "eval_sequential_score": 0.8797917652626597,
      "eval_steps_per_second": 2.62,
      "step": 1400
    },
    {
      "epoch": 9.760201935212454,
      "grad_norm": 4.450948715209961,
      "learning_rate": 1.944438405380829e-05,
      "loss": 16.7047,
      "step": 1450
    },
    {
      "epoch": 9.760201935212454,
      "eval_dim_128_cosine_accuracy": 0.8803833412210128,
      "eval_dim_128_dot_accuracy": 0.12056318031235211,
      "eval_dim_128_euclidean_accuracy": 0.8810932323710364,
      "eval_dim_128_manhattan_accuracy": 0.8802650260293422,
      "eval_dim_128_max_accuracy": 0.8810932323710364,
      "eval_dim_256_cosine_accuracy": 0.8809749171793658,
      "eval_dim_256_dot_accuracy": 0.11914339801230478,
      "eval_dim_256_euclidean_accuracy": 0.8813298627543776,
      "eval_dim_256_manhattan_accuracy": 0.881211547562707,
      "eval_dim_256_max_accuracy": 0.8813298627543776,
      "eval_dim_384_cosine_accuracy": 0.8809749171793658,
      "eval_dim_384_dot_accuracy": 0.11902508282063418,
      "eval_dim_384_euclidean_accuracy": 0.8809749171793658,
      "eval_dim_384_manhattan_accuracy": 0.8820397539044014,
      "eval_dim_384_max_accuracy": 0.8820397539044014,
      "eval_dim_64_cosine_accuracy": 0.8796734500709891,
      "eval_dim_64_dot_accuracy": 0.12245622337908188,
      "eval_dim_64_euclidean_accuracy": 0.880028395646001,
      "eval_dim_64_manhattan_accuracy": 0.8803833412210128,
      "eval_dim_64_max_accuracy": 0.8803833412210128,
      "eval_loss": 16.39591407775879,
      "eval_runtime": 102.018,
      "eval_samples_per_second": 82.848,
      "eval_sequential_score": 0.8796734500709891,
      "eval_steps_per_second": 2.598,
      "step": 1450
    },
    {
      "epoch": 10.096760622633571,
      "grad_norm": 6.13853120803833,
      "learning_rate": 1.9364241599913923e-05,
      "loss": 16.6782,
      "step": 1500
    },
    {
      "epoch": 10.096760622633571,
      "eval_dim_128_cosine_accuracy": 0.8788452437292948,
      "eval_dim_128_dot_accuracy": 0.1228111689540937,
      "eval_dim_128_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_128_manhattan_accuracy": 0.879081874112636,
      "eval_dim_128_max_accuracy": 0.8796734500709891,
      "eval_dim_256_cosine_accuracy": 0.879081874112636,
      "eval_dim_256_dot_accuracy": 0.12091812588736393,
      "eval_dim_256_euclidean_accuracy": 0.8797917652626597,
      "eval_dim_256_manhattan_accuracy": 0.8803833412210128,
      "eval_dim_256_max_accuracy": 0.8803833412210128,
      "eval_dim_384_cosine_accuracy": 0.8795551348793185,
      "eval_dim_384_dot_accuracy": 0.12044486512068149,
      "eval_dim_384_euclidean_accuracy": 0.8795551348793185,
      "eval_dim_384_manhattan_accuracy": 0.8799100804543304,
      "eval_dim_384_max_accuracy": 0.8799100804543304,
      "eval_dim_64_cosine_accuracy": 0.8783719829626124,
      "eval_dim_64_dot_accuracy": 0.12363937529578797,
      "eval_dim_64_euclidean_accuracy": 0.8795551348793185,
      "eval_dim_64_manhattan_accuracy": 0.8781353525792712,
      "eval_dim_64_max_accuracy": 0.8795551348793185,
      "eval_loss": 16.398588180541992,
      "eval_runtime": 103.6429,
      "eval_samples_per_second": 81.549,
      "eval_sequential_score": 0.8783719829626124,
      "eval_steps_per_second": 2.557,
      "step": 1500
    },
    {
      "epoch": 10.43331931005469,
      "grad_norm": 4.757913112640381,
      "learning_rate": 1.9278890272965097e-05,
      "loss": 16.6708,
      "step": 1550
    },
    {
      "epoch": 10.43331931005469,
      "eval_dim_128_cosine_accuracy": 0.8794368196876479,
      "eval_dim_128_dot_accuracy": 0.121509701845717,
      "eval_dim_128_euclidean_accuracy": 0.8795551348793185,
      "eval_dim_128_manhattan_accuracy": 0.879081874112636,
      "eval_dim_128_max_accuracy": 0.8795551348793185,
      "eval_dim_256_cosine_accuracy": 0.8792001893043067,
      "eval_dim_256_dot_accuracy": 0.11961665877898722,
      "eval_dim_256_euclidean_accuracy": 0.8794368196876479,
      "eval_dim_256_manhattan_accuracy": 0.8795551348793185,
      "eval_dim_256_max_accuracy": 0.8795551348793185,
      "eval_dim_384_cosine_accuracy": 0.8796734500709891,
      "eval_dim_384_dot_accuracy": 0.12032654992901089,
      "eval_dim_384_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_384_manhattan_accuracy": 0.8809749171793658,
      "eval_dim_384_max_accuracy": 0.8809749171793658,
      "eval_dim_64_cosine_accuracy": 0.879081874112636,
      "eval_dim_64_dot_accuracy": 0.12245622337908188,
      "eval_dim_64_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_64_manhattan_accuracy": 0.8802650260293422,
      "eval_dim_64_max_accuracy": 0.8802650260293422,
      "eval_loss": 16.401565551757812,
      "eval_runtime": 103.0896,
      "eval_samples_per_second": 81.987,
      "eval_sequential_score": 0.879081874112636,
      "eval_steps_per_second": 2.571,
      "step": 1550
    },
    {
      "epoch": 10.76987799747581,
      "grad_norm": 5.452834129333496,
      "learning_rate": 1.9188377549761962e-05,
      "loss": 16.6485,
      "step": 1600
    },
    {
      "epoch": 10.76987799747581,
      "eval_dim_128_cosine_accuracy": 0.8789635589209654,
      "eval_dim_128_dot_accuracy": 0.1216280170373876,
      "eval_dim_128_euclidean_accuracy": 0.8789635589209654,
      "eval_dim_128_manhattan_accuracy": 0.8781353525792712,
      "eval_dim_128_max_accuracy": 0.8789635589209654,
      "eval_dim_256_cosine_accuracy": 0.8801467108376716,
      "eval_dim_256_dot_accuracy": 0.11985328916232844,
      "eval_dim_256_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_256_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_256_max_accuracy": 0.8801467108376716,
      "eval_dim_384_cosine_accuracy": 0.879081874112636,
      "eval_dim_384_dot_accuracy": 0.12091812588736393,
      "eval_dim_384_euclidean_accuracy": 0.879081874112636,
      "eval_dim_384_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_384_max_accuracy": 0.8794368196876479,
      "eval_dim_64_cosine_accuracy": 0.8781353525792712,
      "eval_dim_64_dot_accuracy": 0.12304779933743493,
      "eval_dim_64_euclidean_accuracy": 0.8783719829626124,
      "eval_dim_64_manhattan_accuracy": 0.879081874112636,
      "eval_dim_64_max_accuracy": 0.879081874112636,
      "eval_loss": 16.396345138549805,
      "eval_runtime": 103.471,
      "eval_samples_per_second": 81.685,
      "eval_sequential_score": 0.8781353525792712,
      "eval_steps_per_second": 2.561,
      "step": 1600
    },
    {
      "epoch": 11.106436684896929,
      "grad_norm": 3.5591487884521484,
      "learning_rate": 1.9092753778138885e-05,
      "loss": 16.6205,
      "step": 1650
    },
    {
      "epoch": 11.106436684896929,
      "eval_dim_128_cosine_accuracy": 0.8778987221959299,
      "eval_dim_128_dot_accuracy": 0.12316611452910553,
      "eval_dim_128_euclidean_accuracy": 0.8781353525792712,
      "eval_dim_128_manhattan_accuracy": 0.8780170373876006,
      "eval_dim_128_max_accuracy": 0.8781353525792712,
      "eval_dim_256_cosine_accuracy": 0.8787269285376242,
      "eval_dim_256_dot_accuracy": 0.121509701845717,
      "eval_dim_256_euclidean_accuracy": 0.8787269285376242,
      "eval_dim_256_manhattan_accuracy": 0.8793185044959773,
      "eval_dim_256_max_accuracy": 0.8793185044959773,
      "eval_dim_384_cosine_accuracy": 0.8793185044959773,
      "eval_dim_384_dot_accuracy": 0.12068149550402271,
      "eval_dim_384_euclidean_accuracy": 0.8793185044959773,
      "eval_dim_384_manhattan_accuracy": 0.8801467108376716,
      "eval_dim_384_max_accuracy": 0.8801467108376716,
      "eval_dim_64_cosine_accuracy": 0.8770705158542357,
      "eval_dim_64_dot_accuracy": 0.12541410317084714,
      "eval_dim_64_euclidean_accuracy": 0.8771888310459063,
      "eval_dim_64_manhattan_accuracy": 0.8776620918125887,
      "eval_dim_64_max_accuracy": 0.8776620918125887,
      "eval_loss": 16.401174545288086,
      "eval_runtime": 102.9169,
      "eval_samples_per_second": 82.124,
      "eval_sequential_score": 0.8770705158542357,
      "eval_steps_per_second": 2.575,
      "step": 1650
    },
    {
      "epoch": 11.442995372318048,
      "grad_norm": 3.712305784225464,
      "learning_rate": 1.8992072148958368e-05,
      "loss": 16.6095,
      "step": 1700
    },
    {
      "epoch": 11.442995372318048,
      "eval_dim_128_cosine_accuracy": 0.8786086133459536,
      "eval_dim_128_dot_accuracy": 0.12233790818741126,
      "eval_dim_128_euclidean_accuracy": 0.878490298154283,
      "eval_dim_128_manhattan_accuracy": 0.8786086133459536,
      "eval_dim_128_max_accuracy": 0.8786086133459536,
      "eval_dim_256_cosine_accuracy": 0.8789635589209654,
      "eval_dim_256_dot_accuracy": 0.1216280170373876,
      "eval_dim_256_euclidean_accuracy": 0.879081874112636,
      "eval_dim_256_manhattan_accuracy": 0.8780170373876006,
      "eval_dim_256_max_accuracy": 0.879081874112636,
      "eval_dim_384_cosine_accuracy": 0.8794368196876479,
      "eval_dim_384_dot_accuracy": 0.12056318031235211,
      "eval_dim_384_euclidean_accuracy": 0.8794368196876479,
      "eval_dim_384_manhattan_accuracy": 0.879081874112636,
      "eval_dim_384_max_accuracy": 0.8794368196876479,
      "eval_dim_64_cosine_accuracy": 0.879081874112636,
      "eval_dim_64_dot_accuracy": 0.12541410317084714,
      "eval_dim_64_euclidean_accuracy": 0.8777804070042593,
      "eval_dim_64_manhattan_accuracy": 0.8788452437292948,
      "eval_dim_64_max_accuracy": 0.879081874112636,
      "eval_loss": 16.413122177124023,
      "eval_runtime": 103.5898,
      "eval_samples_per_second": 81.591,
      "eval_sequential_score": 0.879081874112636,
      "eval_steps_per_second": 2.558,
      "step": 1700
    },
    {
      "epoch": 11.779554059739167,
      "grad_norm": 4.9205145835876465,
      "learning_rate": 1.888638866652356e-05,
      "loss": 16.5891,
      "step": 1750
    },
    {
      "epoch": 11.779554059739167,
      "eval_dim_128_cosine_accuracy": 0.8807382867960246,
      "eval_dim_128_dot_accuracy": 0.1194983435873166,
      "eval_dim_128_euclidean_accuracy": 0.8805016564126834,
      "eval_dim_128_manhattan_accuracy": 0.8792001893043067,
      "eval_dim_128_max_accuracy": 0.8807382867960246,
      "eval_dim_256_cosine_accuracy": 0.8805016564126834,
      "eval_dim_256_dot_accuracy": 0.11902508282063418,
      "eval_dim_256_euclidean_accuracy": 0.8797917652626597,
      "eval_dim_256_manhattan_accuracy": 0.8795551348793185,
      "eval_dim_256_max_accuracy": 0.8805016564126834,
      "eval_dim_384_cosine_accuracy": 0.8809749171793658,
      "eval_dim_384_dot_accuracy": 0.11902508282063418,
      "eval_dim_384_euclidean_accuracy": 0.8809749171793658,
      "eval_dim_384_manhattan_accuracy": 0.880028395646001,
      "eval_dim_384_max_accuracy": 0.8809749171793658,
      "eval_dim_64_cosine_accuracy": 0.8801467108376716,
      "eval_dim_64_dot_accuracy": 0.12292948414576432,
      "eval_dim_64_euclidean_accuracy": 0.879081874112636,
      "eval_dim_64_manhattan_accuracy": 0.879081874112636,
      "eval_dim_64_max_accuracy": 0.8801467108376716,
      "eval_loss": 16.40700340270996,
      "eval_runtime": 103.5887,
      "eval_samples_per_second": 81.592,
      "eval_sequential_score": 0.8801467108376716,
      "eval_steps_per_second": 2.558,
      "step": 1750
    },
    {
      "epoch": 12.116112747160287,
      "grad_norm": 4.849546909332275,
      "learning_rate": 1.8775762117425777e-05,
      "loss": 16.5619,
      "step": 1800
    },
    {
      "epoch": 12.116112747160287,
      "eval_dim_128_cosine_accuracy": 0.8794368196876479,
      "eval_dim_128_dot_accuracy": 0.121509701845717,
      "eval_dim_128_euclidean_accuracy": 0.8792001893043067,
      "eval_dim_128_manhattan_accuracy": 0.8789635589209654,
      "eval_dim_128_max_accuracy": 0.8794368196876479,
      "eval_dim_256_cosine_accuracy": 0.880028395646001,
      "eval_dim_256_dot_accuracy": 0.11973497397065783,
      "eval_dim_256_euclidean_accuracy": 0.8799100804543304,
      "eval_dim_256_manhattan_accuracy": 0.8792001893043067,
      "eval_dim_256_max_accuracy": 0.880028395646001,
      "eval_dim_384_cosine_accuracy": 0.8796734500709891,
      "eval_dim_384_dot_accuracy": 0.12032654992901089,
      "eval_dim_384_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_384_manhattan_accuracy": 0.8796734500709891,
      "eval_dim_384_max_accuracy": 0.8796734500709891,
      "eval_dim_64_cosine_accuracy": 0.8780170373876006,
      "eval_dim_64_dot_accuracy": 0.12470421202082348,
      "eval_dim_64_euclidean_accuracy": 0.8797917652626597,
      "eval_dim_64_manhattan_accuracy": 0.8786086133459536,
      "eval_dim_64_max_accuracy": 0.8797917652626597,
      "eval_loss": 16.396265029907227,
      "eval_runtime": 102.3506,
      "eval_samples_per_second": 82.579,
      "eval_sequential_score": 0.8780170373876006,
      "eval_steps_per_second": 2.589,
      "step": 1800
    },
    {
      "epoch": 12.452671434581404,
      "grad_norm": 4.944924831390381,
      "learning_rate": 1.866025403784439e-05,
      "loss": 16.5467,
      "step": 1850
    },
    {
      "epoch": 12.452671434581404,
      "eval_dim_128_cosine_accuracy": 0.8795551348793185,
      "eval_dim_128_dot_accuracy": 0.12316611452910553,
      "eval_dim_128_euclidean_accuracy": 0.8787269285376242,
      "eval_dim_128_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_128_max_accuracy": 0.8795551348793185,
      "eval_dim_256_cosine_accuracy": 0.880619971604354,
      "eval_dim_256_dot_accuracy": 0.12068149550402271,
      "eval_dim_256_euclidean_accuracy": 0.8794368196876479,
      "eval_dim_256_manhattan_accuracy": 0.8801467108376716,
      "eval_dim_256_max_accuracy": 0.880619971604354,
      "eval_dim_384_cosine_accuracy": 0.8803833412210128,
      "eval_dim_384_dot_accuracy": 0.11961665877898722,
      "eval_dim_384_euclidean_accuracy": 0.8803833412210128,
      "eval_dim_384_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_384_max_accuracy": 0.8807382867960246,
      "eval_dim_64_cosine_accuracy": 0.8789635589209654,
      "eval_dim_64_dot_accuracy": 0.12470421202082348,
      "eval_dim_64_euclidean_accuracy": 0.8781353525792712,
      "eval_dim_64_manhattan_accuracy": 0.8796734500709891,
      "eval_dim_64_max_accuracy": 0.8796734500709891,
      "eval_loss": 16.399133682250977,
      "eval_runtime": 104.1432,
      "eval_samples_per_second": 81.157,
      "eval_sequential_score": 0.8789635589209654,
      "eval_steps_per_second": 2.545,
      "step": 1850
    },
    {
      "epoch": 12.789230122002524,
      "grad_norm": 6.032313346862793,
      "learning_rate": 1.853992867931721e-05,
      "loss": 16.5398,
      "step": 1900
    },
    {
      "epoch": 12.789230122002524,
      "eval_dim_128_cosine_accuracy": 0.8792001893043067,
      "eval_dim_128_dot_accuracy": 0.12139138665404638,
      "eval_dim_128_euclidean_accuracy": 0.8797917652626597,
      "eval_dim_128_manhattan_accuracy": 0.8787269285376242,
      "eval_dim_128_max_accuracy": 0.8797917652626597,
      "eval_dim_256_cosine_accuracy": 0.8797917652626597,
      "eval_dim_256_dot_accuracy": 0.11973497397065783,
      "eval_dim_256_euclidean_accuracy": 0.8797917652626597,
      "eval_dim_256_manhattan_accuracy": 0.8792001893043067,
      "eval_dim_256_max_accuracy": 0.8797917652626597,
      "eval_dim_384_cosine_accuracy": 0.8801467108376716,
      "eval_dim_384_dot_accuracy": 0.11985328916232844,
      "eval_dim_384_euclidean_accuracy": 0.8801467108376716,
      "eval_dim_384_manhattan_accuracy": 0.8805016564126834,
      "eval_dim_384_max_accuracy": 0.8805016564126834,
      "eval_dim_64_cosine_accuracy": 0.8788452437292948,
      "eval_dim_64_dot_accuracy": 0.12423095125414103,
      "eval_dim_64_euclidean_accuracy": 0.8793185044959773,
      "eval_dim_64_manhattan_accuracy": 0.8776620918125887,
      "eval_dim_64_max_accuracy": 0.8793185044959773,
      "eval_loss": 16.397045135498047,
      "eval_runtime": 103.5361,
      "eval_samples_per_second": 81.633,
      "eval_sequential_score": 0.8788452437292948,
      "eval_steps_per_second": 2.559,
      "step": 1900
    },
    {
      "epoch": 13.125788809423643,
      "grad_norm": 4.27797269821167,
      "learning_rate": 1.8414852973000503e-05,
      "loss": 16.5047,
      "step": 1950
    },
    {
      "epoch": 13.125788809423643,
      "eval_dim_128_cosine_accuracy": 0.8795551348793185,
      "eval_dim_128_dot_accuracy": 0.1216280170373876,
      "eval_dim_128_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_128_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_128_max_accuracy": 0.8797917652626597,
      "eval_dim_256_cosine_accuracy": 0.8803833412210128,
      "eval_dim_256_dot_accuracy": 0.12068149550402271,
      "eval_dim_256_euclidean_accuracy": 0.8797917652626597,
      "eval_dim_256_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_256_max_accuracy": 0.8803833412210128,
      "eval_dim_384_cosine_accuracy": 0.8803833412210128,
      "eval_dim_384_dot_accuracy": 0.11961665877898722,
      "eval_dim_384_euclidean_accuracy": 0.8803833412210128,
      "eval_dim_384_manhattan_accuracy": 0.8805016564126834,
      "eval_dim_384_max_accuracy": 0.8805016564126834,
      "eval_dim_64_cosine_accuracy": 0.8788452437292948,
      "eval_dim_64_dot_accuracy": 0.12588736393752958,
      "eval_dim_64_euclidean_accuracy": 0.8793185044959773,
      "eval_dim_64_manhattan_accuracy": 0.8802650260293422,
      "eval_dim_64_max_accuracy": 0.8802650260293422,
      "eval_loss": 16.396381378173828,
      "eval_runtime": 102.672,
      "eval_samples_per_second": 82.32,
      "eval_sequential_score": 0.8788452437292948,
      "eval_steps_per_second": 2.581,
      "step": 1950
    },
    {
      "epoch": 13.462347496844762,
      "grad_norm": 4.051229953765869,
      "learning_rate": 1.8285096492438424e-05,
      "loss": 16.4985,
      "step": 2000
    },
    {
      "epoch": 13.462347496844762,
      "eval_dim_128_cosine_accuracy": 0.8793185044959773,
      "eval_dim_128_dot_accuracy": 0.12127307146237577,
      "eval_dim_128_euclidean_accuracy": 0.8803833412210128,
      "eval_dim_128_manhattan_accuracy": 0.8796734500709891,
      "eval_dim_128_max_accuracy": 0.8803833412210128,
      "eval_dim_256_cosine_accuracy": 0.8797917652626597,
      "eval_dim_256_dot_accuracy": 0.12020823473734027,
      "eval_dim_256_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_256_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_256_max_accuracy": 0.8797917652626597,
      "eval_dim_384_cosine_accuracy": 0.8807382867960246,
      "eval_dim_384_dot_accuracy": 0.11926171320397538,
      "eval_dim_384_euclidean_accuracy": 0.8807382867960246,
      "eval_dim_384_manhattan_accuracy": 0.8810932323710364,
      "eval_dim_384_max_accuracy": 0.8810932323710364,
      "eval_dim_64_cosine_accuracy": 0.8789635589209654,
      "eval_dim_64_dot_accuracy": 0.12316611452910553,
      "eval_dim_64_euclidean_accuracy": 0.8787269285376242,
      "eval_dim_64_manhattan_accuracy": 0.879081874112636,
      "eval_dim_64_max_accuracy": 0.879081874112636,
      "eval_loss": 16.4024600982666,
      "eval_runtime": 104.2185,
      "eval_samples_per_second": 81.099,
      "eval_sequential_score": 0.8789635589209654,
      "eval_steps_per_second": 2.543,
      "step": 2000
    },
    {
      "epoch": 13.798906184265881,
      "grad_norm": 4.3837666511535645,
      "learning_rate": 1.8150731414862623e-05,
      "loss": 16.4852,
      "step": 2050
    },
    {
      "epoch": 13.798906184265881,
      "eval_dim_128_cosine_accuracy": 0.8801467108376716,
      "eval_dim_128_dot_accuracy": 0.12032654992901089,
      "eval_dim_128_euclidean_accuracy": 0.8805016564126834,
      "eval_dim_128_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_128_max_accuracy": 0.8805016564126834,
      "eval_dim_256_cosine_accuracy": 0.8809749171793658,
      "eval_dim_256_dot_accuracy": 0.119380028395646,
      "eval_dim_256_euclidean_accuracy": 0.8814481779460482,
      "eval_dim_256_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_256_max_accuracy": 0.8814481779460482,
      "eval_dim_384_cosine_accuracy": 0.880028395646001,
      "eval_dim_384_dot_accuracy": 0.11997160435399905,
      "eval_dim_384_euclidean_accuracy": 0.880028395646001,
      "eval_dim_384_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_384_max_accuracy": 0.880028395646001,
      "eval_dim_64_cosine_accuracy": 0.8793185044959773,
      "eval_dim_64_dot_accuracy": 0.12352106010411737,
      "eval_dim_64_euclidean_accuracy": 0.8801467108376716,
      "eval_dim_64_manhattan_accuracy": 0.8796734500709891,
      "eval_dim_64_max_accuracy": 0.8801467108376716,
      "eval_loss": 16.410737991333008,
      "eval_runtime": 102.5333,
      "eval_samples_per_second": 82.432,
      "eval_sequential_score": 0.8793185044959773,
      "eval_steps_per_second": 2.585,
      "step": 2050
    },
    {
      "epoch": 14.135464871687,
      "grad_norm": 4.87747859954834,
      "learning_rate": 1.8011832481043577e-05,
      "loss": 16.4526,
      "step": 2100
    },
    {
      "epoch": 14.135464871687,
      "eval_dim_128_cosine_accuracy": 0.8796734500709891,
      "eval_dim_128_dot_accuracy": 0.12103644107903455,
      "eval_dim_128_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_128_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_128_max_accuracy": 0.8796734500709891,
      "eval_dim_256_cosine_accuracy": 0.8801467108376716,
      "eval_dim_256_dot_accuracy": 0.12068149550402271,
      "eval_dim_256_euclidean_accuracy": 0.8805016564126834,
      "eval_dim_256_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_256_max_accuracy": 0.8805016564126834,
      "eval_dim_384_cosine_accuracy": 0.8808566019876952,
      "eval_dim_384_dot_accuracy": 0.11914339801230478,
      "eval_dim_384_euclidean_accuracy": 0.8808566019876952,
      "eval_dim_384_manhattan_accuracy": 0.8810932323710364,
      "eval_dim_384_max_accuracy": 0.8810932323710364,
      "eval_dim_64_cosine_accuracy": 0.8778987221959299,
      "eval_dim_64_dot_accuracy": 0.12470421202082348,
      "eval_dim_64_euclidean_accuracy": 0.8781353525792712,
      "eval_dim_64_manhattan_accuracy": 0.879081874112636,
      "eval_dim_64_max_accuracy": 0.879081874112636,
      "eval_loss": 16.392879486083984,
      "eval_runtime": 103.5589,
      "eval_samples_per_second": 81.615,
      "eval_sequential_score": 0.8778987221959299,
      "eval_steps_per_second": 2.559,
      "step": 2100
    },
    {
      "epoch": 14.47202355910812,
      "grad_norm": 6.463150978088379,
      "learning_rate": 1.78684769537159e-05,
      "loss": 16.4343,
      "step": 2150
    },
    {
      "epoch": 14.47202355910812,
      "eval_dim_128_cosine_accuracy": 0.8788452437292948,
      "eval_dim_128_dot_accuracy": 0.12221959299574066,
      "eval_dim_128_euclidean_accuracy": 0.878490298154283,
      "eval_dim_128_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_128_max_accuracy": 0.8797917652626597,
      "eval_dim_256_cosine_accuracy": 0.879081874112636,
      "eval_dim_256_dot_accuracy": 0.121509701845717,
      "eval_dim_256_euclidean_accuracy": 0.8786086133459536,
      "eval_dim_256_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_256_max_accuracy": 0.8797917652626597,
      "eval_dim_384_cosine_accuracy": 0.8796734500709891,
      "eval_dim_384_dot_accuracy": 0.12032654992901089,
      "eval_dim_384_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_384_manhattan_accuracy": 0.880619971604354,
      "eval_dim_384_max_accuracy": 0.880619971604354,
      "eval_dim_64_cosine_accuracy": 0.8774254614292475,
      "eval_dim_64_dot_accuracy": 0.1250591575958353,
      "eval_dim_64_euclidean_accuracy": 0.8774254614292475,
      "eval_dim_64_manhattan_accuracy": 0.8780170373876006,
      "eval_dim_64_max_accuracy": 0.8780170373876006,
      "eval_loss": 16.40749740600586,
      "eval_runtime": 102.9532,
      "eval_samples_per_second": 82.096,
      "eval_sequential_score": 0.8774254614292475,
      "eval_steps_per_second": 2.574,
      "step": 2150
    },
    {
      "epoch": 14.80858224652924,
      "grad_norm": 4.839356422424316,
      "learning_rate": 1.7720744574600865e-05,
      "loss": 16.4244,
      "step": 2200
    },
    {
      "epoch": 14.80858224652924,
      "eval_dim_128_cosine_accuracy": 0.8803833412210128,
      "eval_dim_128_dot_accuracy": 0.11973497397065783,
      "eval_dim_128_euclidean_accuracy": 0.880619971604354,
      "eval_dim_128_manhattan_accuracy": 0.8809749171793658,
      "eval_dim_128_max_accuracy": 0.8809749171793658,
      "eval_dim_256_cosine_accuracy": 0.8819214387127308,
      "eval_dim_256_dot_accuracy": 0.119380028395646,
      "eval_dim_256_euclidean_accuracy": 0.8815664931377188,
      "eval_dim_256_manhattan_accuracy": 0.8814481779460482,
      "eval_dim_256_max_accuracy": 0.8819214387127308,
      "eval_dim_384_cosine_accuracy": 0.8820397539044014,
      "eval_dim_384_dot_accuracy": 0.11796024609559867,
      "eval_dim_384_euclidean_accuracy": 0.8820397539044014,
      "eval_dim_384_manhattan_accuracy": 0.882158069096072,
      "eval_dim_384_max_accuracy": 0.882158069096072,
      "eval_dim_64_cosine_accuracy": 0.8808566019876952,
      "eval_dim_64_dot_accuracy": 0.12221959299574066,
      "eval_dim_64_euclidean_accuracy": 0.880619971604354,
      "eval_dim_64_manhattan_accuracy": 0.8786086133459536,
      "eval_dim_64_max_accuracy": 0.8808566019876952,
      "eval_loss": 16.402673721313477,
      "eval_runtime": 103.4179,
      "eval_samples_per_second": 81.727,
      "eval_sequential_score": 0.8808566019876952,
      "eval_steps_per_second": 2.562,
      "step": 2200
    },
    {
      "epoch": 15.145140933950358,
      "grad_norm": 5.812349796295166,
      "learning_rate": 1.756871752004992e-05,
      "loss": 16.3947,
      "step": 2250
    },
    {
      "epoch": 15.145140933950358,
      "eval_dim_128_cosine_accuracy": 0.879081874112636,
      "eval_dim_128_dot_accuracy": 0.12316611452910553,
      "eval_dim_128_euclidean_accuracy": 0.8786086133459536,
      "eval_dim_128_manhattan_accuracy": 0.8809749171793658,
      "eval_dim_128_max_accuracy": 0.8809749171793658,
      "eval_dim_256_cosine_accuracy": 0.8792001893043067,
      "eval_dim_256_dot_accuracy": 0.12139138665404638,
      "eval_dim_256_euclidean_accuracy": 0.8801467108376716,
      "eval_dim_256_manhattan_accuracy": 0.8813298627543776,
      "eval_dim_256_max_accuracy": 0.8813298627543776,
      "eval_dim_384_cosine_accuracy": 0.8802650260293422,
      "eval_dim_384_dot_accuracy": 0.11973497397065783,
      "eval_dim_384_euclidean_accuracy": 0.8802650260293422,
      "eval_dim_384_manhattan_accuracy": 0.8808566019876952,
      "eval_dim_384_max_accuracy": 0.8808566019876952,
      "eval_dim_64_cosine_accuracy": 0.8773071462375769,
      "eval_dim_64_dot_accuracy": 0.12695220066256507,
      "eval_dim_64_euclidean_accuracy": 0.8768338854708945,
      "eval_dim_64_manhattan_accuracy": 0.8787269285376242,
      "eval_dim_64_max_accuracy": 0.8787269285376242,
      "eval_loss": 16.4101619720459,
      "eval_runtime": 105.1832,
      "eval_samples_per_second": 80.355,
      "eval_sequential_score": 0.8773071462375769,
      "eval_steps_per_second": 2.519,
      "step": 2250
    },
    {
      "epoch": 15.481699621371476,
      "grad_norm": 4.386394023895264,
      "learning_rate": 1.7412480355334006e-05,
      "loss": 16.3827,
      "step": 2300
    },
    {
      "epoch": 15.481699621371476,
      "eval_dim_128_cosine_accuracy": 0.8803833412210128,
      "eval_dim_128_dot_accuracy": 0.12245622337908188,
      "eval_dim_128_euclidean_accuracy": 0.880619971604354,
      "eval_dim_128_manhattan_accuracy": 0.88180312352106,
      "eval_dim_128_max_accuracy": 0.88180312352106,
      "eval_dim_256_cosine_accuracy": 0.8813298627543776,
      "eval_dim_256_dot_accuracy": 0.12079981069569333,
      "eval_dim_256_euclidean_accuracy": 0.8809749171793658,
      "eval_dim_256_manhattan_accuracy": 0.8819214387127308,
      "eval_dim_256_max_accuracy": 0.8819214387127308,
      "eval_dim_384_cosine_accuracy": 0.8813298627543776,
      "eval_dim_384_dot_accuracy": 0.11867013724562234,
      "eval_dim_384_euclidean_accuracy": 0.8813298627543776,
      "eval_dim_384_manhattan_accuracy": 0.8809749171793658,
      "eval_dim_384_max_accuracy": 0.8813298627543776,
      "eval_dim_64_cosine_accuracy": 0.8781353525792712,
      "eval_dim_64_dot_accuracy": 0.1260056791292002,
      "eval_dim_64_euclidean_accuracy": 0.8789635589209654,
      "eval_dim_64_manhattan_accuracy": 0.8814481779460482,
      "eval_dim_64_max_accuracy": 0.8814481779460482,
      "eval_loss": 16.404207229614258,
      "eval_runtime": 101.3893,
      "eval_samples_per_second": 83.362,
      "eval_sequential_score": 0.8781353525792712,
      "eval_steps_per_second": 2.614,
      "step": 2300
    },
    {
      "epoch": 15.818258308792595,
      "grad_norm": 4.8762359619140625,
      "learning_rate": 1.7252119987603976e-05,
      "loss": 16.3719,
      "step": 2350
    },
    {
      "epoch": 15.818258308792595,
      "eval_dim_128_cosine_accuracy": 0.8801467108376716,
      "eval_dim_128_dot_accuracy": 0.12032654992901089,
      "eval_dim_128_euclidean_accuracy": 0.8802650260293422,
      "eval_dim_128_manhattan_accuracy": 0.8802650260293422,
      "eval_dim_128_max_accuracy": 0.8802650260293422,
      "eval_dim_256_cosine_accuracy": 0.88180312352106,
      "eval_dim_256_dot_accuracy": 0.11878845243729295,
      "eval_dim_256_euclidean_accuracy": 0.8820397539044014,
      "eval_dim_256_manhattan_accuracy": 0.881211547562707,
      "eval_dim_256_max_accuracy": 0.8820397539044014,
      "eval_dim_384_cosine_accuracy": 0.8820397539044014,
      "eval_dim_384_dot_accuracy": 0.11796024609559867,
      "eval_dim_384_euclidean_accuracy": 0.8820397539044014,
      "eval_dim_384_manhattan_accuracy": 0.8808566019876952,
      "eval_dim_384_max_accuracy": 0.8820397539044014,
      "eval_dim_64_cosine_accuracy": 0.879081874112636,
      "eval_dim_64_dot_accuracy": 0.12458589682915286,
      "eval_dim_64_euclidean_accuracy": 0.8805016564126834,
      "eval_dim_64_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_64_max_accuracy": 0.8805016564126834,
      "eval_loss": 16.40033721923828,
      "eval_runtime": 104.1264,
      "eval_samples_per_second": 81.171,
      "eval_sequential_score": 0.879081874112636,
      "eval_steps_per_second": 2.545,
      "step": 2350
    },
    {
      "epoch": 16.154816996213714,
      "grad_norm": 5.414395809173584,
      "learning_rate": 1.7087725617548385e-05,
      "loss": 16.3403,
      "step": 2400
    },
    {
      "epoch": 16.154816996213714,
      "eval_dim_128_cosine_accuracy": 0.8781353525792712,
      "eval_dim_128_dot_accuracy": 0.12328442972077615,
      "eval_dim_128_euclidean_accuracy": 0.8778987221959299,
      "eval_dim_128_manhattan_accuracy": 0.8787269285376242,
      "eval_dim_128_max_accuracy": 0.8787269285376242,
      "eval_dim_256_cosine_accuracy": 0.879081874112636,
      "eval_dim_256_dot_accuracy": 0.12210127780407004,
      "eval_dim_256_euclidean_accuracy": 0.8788452437292948,
      "eval_dim_256_manhattan_accuracy": 0.879081874112636,
      "eval_dim_256_max_accuracy": 0.879081874112636,
      "eval_dim_384_cosine_accuracy": 0.8799100804543304,
      "eval_dim_384_dot_accuracy": 0.12008991954566967,
      "eval_dim_384_euclidean_accuracy": 0.8799100804543304,
      "eval_dim_384_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_384_max_accuracy": 0.8799100804543304,
      "eval_dim_64_cosine_accuracy": 0.8767155702792239,
      "eval_dim_64_dot_accuracy": 0.12766209181258872,
      "eval_dim_64_euclidean_accuracy": 0.8765972550875533,
      "eval_dim_64_manhattan_accuracy": 0.8769522006625651,
      "eval_dim_64_max_accuracy": 0.8769522006625651,
      "eval_loss": 16.413236618041992,
      "eval_runtime": 105.1626,
      "eval_samples_per_second": 80.371,
      "eval_sequential_score": 0.8767155702792239,
      "eval_steps_per_second": 2.52,
      "step": 2400
    },
    {
      "epoch": 16.491375683634836,
      "grad_norm": 4.138753414154053,
      "learning_rate": 1.6919388689775463e-05,
      "loss": 16.3357,
      "step": 2450
    },
    {
      "epoch": 16.491375683634836,
      "eval_dim_128_cosine_accuracy": 0.8803833412210128,
      "eval_dim_128_dot_accuracy": 0.1216280170373876,
      "eval_dim_128_euclidean_accuracy": 0.879081874112636,
      "eval_dim_128_manhattan_accuracy": 0.8802650260293422,
      "eval_dim_128_max_accuracy": 0.8803833412210128,
      "eval_dim_256_cosine_accuracy": 0.8808566019876952,
      "eval_dim_256_dot_accuracy": 0.121509701845717,
      "eval_dim_256_euclidean_accuracy": 0.8802650260293422,
      "eval_dim_256_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_256_max_accuracy": 0.8808566019876952,
      "eval_dim_384_cosine_accuracy": 0.8807382867960246,
      "eval_dim_384_dot_accuracy": 0.11926171320397538,
      "eval_dim_384_euclidean_accuracy": 0.8807382867960246,
      "eval_dim_384_manhattan_accuracy": 0.8801467108376716,
      "eval_dim_384_max_accuracy": 0.8807382867960246,
      "eval_dim_64_cosine_accuracy": 0.8792001893043067,
      "eval_dim_64_dot_accuracy": 0.12647893989588263,
      "eval_dim_64_euclidean_accuracy": 0.8786086133459536,
      "eval_dim_64_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_64_max_accuracy": 0.8807382867960246,
      "eval_loss": 16.414878845214844,
      "eval_runtime": 100.6398,
      "eval_samples_per_second": 83.983,
      "eval_sequential_score": 0.8792001893043067,
      "eval_steps_per_second": 2.633,
      "step": 2450
    },
    {
      "epoch": 16.827934371055953,
      "grad_norm": 4.080146312713623,
      "learning_rate": 1.6747202841946928e-05,
      "loss": 16.3203,
      "step": 2500
    },
    {
      "epoch": 16.827934371055953,
      "eval_dim_128_cosine_accuracy": 0.8803833412210128,
      "eval_dim_128_dot_accuracy": 0.12186464742072882,
      "eval_dim_128_euclidean_accuracy": 0.8814481779460482,
      "eval_dim_128_manhattan_accuracy": 0.880619971604354,
      "eval_dim_128_max_accuracy": 0.8814481779460482,
      "eval_dim_256_cosine_accuracy": 0.8814481779460482,
      "eval_dim_256_dot_accuracy": 0.12044486512068149,
      "eval_dim_256_euclidean_accuracy": 0.8820397539044014,
      "eval_dim_256_manhattan_accuracy": 0.8825130146710838,
      "eval_dim_256_max_accuracy": 0.8825130146710838,
      "eval_dim_384_cosine_accuracy": 0.8815664931377188,
      "eval_dim_384_dot_accuracy": 0.11843350686228112,
      "eval_dim_384_euclidean_accuracy": 0.8815664931377188,
      "eval_dim_384_manhattan_accuracy": 0.8815664931377188,
      "eval_dim_384_max_accuracy": 0.8815664931377188,
      "eval_dim_64_cosine_accuracy": 0.879081874112636,
      "eval_dim_64_dot_accuracy": 0.12565073355418835,
      "eval_dim_64_euclidean_accuracy": 0.880028395646001,
      "eval_dim_64_manhattan_accuracy": 0.8810932323710364,
      "eval_dim_64_max_accuracy": 0.8810932323710364,
      "eval_loss": 16.408126831054688,
      "eval_runtime": 103.4973,
      "eval_samples_per_second": 81.664,
      "eval_sequential_score": 0.879081874112636,
      "eval_steps_per_second": 2.56,
      "step": 2500
    },
    {
      "epoch": 17.16449305847707,
      "grad_norm": 5.26322078704834,
      "learning_rate": 1.6571263852691887e-05,
      "loss": 16.2986,
      "step": 2550
    },
    {
      "epoch": 17.16449305847707,
      "eval_dim_128_cosine_accuracy": 0.8797917652626597,
      "eval_dim_128_dot_accuracy": 0.12304779933743493,
      "eval_dim_128_euclidean_accuracy": 0.8801467108376716,
      "eval_dim_128_manhattan_accuracy": 0.8778987221959299,
      "eval_dim_128_max_accuracy": 0.8801467108376716,
      "eval_dim_256_cosine_accuracy": 0.880028395646001,
      "eval_dim_256_dot_accuracy": 0.12068149550402271,
      "eval_dim_256_euclidean_accuracy": 0.880619971604354,
      "eval_dim_256_manhattan_accuracy": 0.8805016564126834,
      "eval_dim_256_max_accuracy": 0.880619971604354,
      "eval_dim_384_cosine_accuracy": 0.8820397539044014,
      "eval_dim_384_dot_accuracy": 0.11796024609559867,
      "eval_dim_384_euclidean_accuracy": 0.8820397539044014,
      "eval_dim_384_manhattan_accuracy": 0.8808566019876952,
      "eval_dim_384_max_accuracy": 0.8820397539044014,
      "eval_dim_64_cosine_accuracy": 0.879081874112636,
      "eval_dim_64_dot_accuracy": 0.12707051585423568,
      "eval_dim_64_euclidean_accuracy": 0.8781353525792712,
      "eval_dim_64_manhattan_accuracy": 0.8778987221959299,
      "eval_dim_64_max_accuracy": 0.879081874112636,
      "eval_loss": 16.413921356201172,
      "eval_runtime": 103.9357,
      "eval_samples_per_second": 81.32,
      "eval_sequential_score": 0.879081874112636,
      "eval_steps_per_second": 2.55,
      "step": 2550
    },
    {
      "epoch": 17.50105174589819,
      "grad_norm": 9.353097915649414,
      "learning_rate": 1.639166958832985e-05,
      "loss": 16.2923,
      "step": 2600
    },
    {
      "epoch": 17.50105174589819,
      "eval_dim_128_cosine_accuracy": 0.8786086133459536,
      "eval_dim_128_dot_accuracy": 0.12352106010411737,
      "eval_dim_128_euclidean_accuracy": 0.8783719829626124,
      "eval_dim_128_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_128_max_accuracy": 0.8807382867960246,
      "eval_dim_256_cosine_accuracy": 0.8792001893043067,
      "eval_dim_256_dot_accuracy": 0.12103644107903455,
      "eval_dim_256_euclidean_accuracy": 0.8796734500709891,
      "eval_dim_256_manhattan_accuracy": 0.8810932323710364,
      "eval_dim_256_max_accuracy": 0.8810932323710364,
      "eval_dim_384_cosine_accuracy": 0.8799100804543304,
      "eval_dim_384_dot_accuracy": 0.12008991954566967,
      "eval_dim_384_euclidean_accuracy": 0.8799100804543304,
      "eval_dim_384_manhattan_accuracy": 0.880028395646001,
      "eval_dim_384_max_accuracy": 0.880028395646001,
      "eval_dim_64_cosine_accuracy": 0.8768338854708945,
      "eval_dim_64_dot_accuracy": 0.12754377662091812,
      "eval_dim_64_euclidean_accuracy": 0.8762423095125415,
      "eval_dim_64_manhattan_accuracy": 0.8789635589209654,
      "eval_dim_64_max_accuracy": 0.8789635589209654,
      "eval_loss": 16.406217575073242,
      "eval_runtime": 101.8719,
      "eval_samples_per_second": 82.967,
      "eval_sequential_score": 0.8768338854708945,
      "eval_steps_per_second": 2.601,
      "step": 2600
    },
    {
      "epoch": 17.83761043331931,
      "grad_norm": 5.8258891105651855,
      "learning_rate": 1.6208519948432438e-05,
      "loss": 16.2649,
      "step": 2650
    },
    {
      "epoch": 17.83761043331931,
      "eval_dim_128_cosine_accuracy": 0.880028395646001,
      "eval_dim_128_dot_accuracy": 0.12186464742072882,
      "eval_dim_128_euclidean_accuracy": 0.8803833412210128,
      "eval_dim_128_manhattan_accuracy": 0.8799100804543304,
      "eval_dim_128_max_accuracy": 0.8803833412210128,
      "eval_dim_256_cosine_accuracy": 0.8807382867960246,
      "eval_dim_256_dot_accuracy": 0.12210127780407004,
      "eval_dim_256_euclidean_accuracy": 0.8814481779460482,
      "eval_dim_256_manhattan_accuracy": 0.8810932323710364,
      "eval_dim_256_max_accuracy": 0.8814481779460482,
      "eval_dim_384_cosine_accuracy": 0.8814481779460482,
      "eval_dim_384_dot_accuracy": 0.11855182205395173,
      "eval_dim_384_euclidean_accuracy": 0.8814481779460482,
      "eval_dim_384_manhattan_accuracy": 0.8814481779460482,
      "eval_dim_384_max_accuracy": 0.8814481779460482,
      "eval_dim_64_cosine_accuracy": 0.8787269285376242,
      "eval_dim_64_dot_accuracy": 0.1283719829626124,
      "eval_dim_64_euclidean_accuracy": 0.8788452437292948,
      "eval_dim_64_manhattan_accuracy": 0.8799100804543304,
      "eval_dim_64_max_accuracy": 0.8799100804543304,
      "eval_loss": 16.410572052001953,
      "eval_runtime": 101.9269,
      "eval_samples_per_second": 82.922,
      "eval_sequential_score": 0.8787269285376242,
      "eval_steps_per_second": 2.6,
      "step": 2650
    },
    {
      "epoch": 18.17416912074043,
      "grad_norm": 4.463468074798584,
      "learning_rate": 1.6021916810254096e-05,
      "loss": 16.2505,
      "step": 2700
    },
    {
      "epoch": 18.17416912074043,
      "eval_dim_128_cosine_accuracy": 0.8786086133459536,
      "eval_dim_128_dot_accuracy": 0.12411263606247042,
      "eval_dim_128_euclidean_accuracy": 0.8780170373876006,
      "eval_dim_128_manhattan_accuracy": 0.8792001893043067,
      "eval_dim_128_max_accuracy": 0.8792001893043067,
      "eval_dim_256_cosine_accuracy": 0.8793185044959773,
      "eval_dim_256_dot_accuracy": 0.12210127780407004,
      "eval_dim_256_euclidean_accuracy": 0.8793185044959773,
      "eval_dim_256_manhattan_accuracy": 0.8805016564126834,
      "eval_dim_256_max_accuracy": 0.8805016564126834,
      "eval_dim_384_cosine_accuracy": 0.8802650260293422,
      "eval_dim_384_dot_accuracy": 0.11973497397065783,
      "eval_dim_384_euclidean_accuracy": 0.8802650260293422,
      "eval_dim_384_manhattan_accuracy": 0.8813298627543776,
      "eval_dim_384_max_accuracy": 0.8813298627543776,
      "eval_dim_64_cosine_accuracy": 0.8770705158542357,
      "eval_dim_64_dot_accuracy": 0.13014671083767157,
      "eval_dim_64_euclidean_accuracy": 0.8761239943208708,
      "eval_dim_64_manhattan_accuracy": 0.8787269285376242,
      "eval_dim_64_max_accuracy": 0.8787269285376242,
      "eval_loss": 16.418752670288086,
      "eval_runtime": 106.398,
      "eval_samples_per_second": 79.438,
      "eval_sequential_score": 0.8770705158542357,
      "eval_steps_per_second": 2.491,
      "step": 2700
    },
    {
      "epoch": 18.510727808161548,
      "grad_norm": 5.066239833831787,
      "learning_rate": 1.5831963972062734e-05,
      "loss": 16.226,
      "step": 2750
    },
    {
      "epoch": 18.510727808161548,
      "eval_dim_128_cosine_accuracy": 0.8770705158542357,
      "eval_dim_128_dot_accuracy": 0.12446758163748226,
      "eval_dim_128_euclidean_accuracy": 0.8771888310459063,
      "eval_dim_128_manhattan_accuracy": 0.8778987221959299,
      "eval_dim_128_max_accuracy": 0.8778987221959299,
      "eval_dim_256_cosine_accuracy": 0.8781353525792712,
      "eval_dim_256_dot_accuracy": 0.12304779933743493,
      "eval_dim_256_euclidean_accuracy": 0.8788452437292948,
      "eval_dim_256_manhattan_accuracy": 0.8799100804543304,
      "eval_dim_256_max_accuracy": 0.8799100804543304,
      "eval_dim_384_cosine_accuracy": 0.8780170373876006,
      "eval_dim_384_dot_accuracy": 0.12198296261239944,
      "eval_dim_384_euclidean_accuracy": 0.8780170373876006,
      "eval_dim_384_manhattan_accuracy": 0.8770705158542357,
      "eval_dim_384_max_accuracy": 0.8780170373876006,
      "eval_dim_64_cosine_accuracy": 0.8765972550875533,
      "eval_dim_64_dot_accuracy": 0.12884524372929484,
      "eval_dim_64_euclidean_accuracy": 0.8765972550875533,
      "eval_dim_64_manhattan_accuracy": 0.8778987221959299,
      "eval_dim_64_max_accuracy": 0.8778987221959299,
      "eval_loss": 16.4149112701416,
      "eval_runtime": 101.2915,
      "eval_samples_per_second": 83.442,
      "eval_sequential_score": 0.8765972550875533,
      "eval_steps_per_second": 2.616,
      "step": 2750
    },
    {
      "epoch": 18.84728649558267,
      "grad_norm": 4.982476234436035,
      "learning_rate": 1.5638767095401778e-05,
      "loss": 16.2106,
      "step": 2800
    },
    {
      "epoch": 18.84728649558267,
      "eval_dim_128_cosine_accuracy": 0.8780170373876006,
      "eval_dim_128_dot_accuracy": 0.12529578797917654,
      "eval_dim_128_euclidean_accuracy": 0.878490298154283,
      "eval_dim_128_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_128_max_accuracy": 0.8794368196876479,
      "eval_dim_256_cosine_accuracy": 0.8799100804543304,
      "eval_dim_256_dot_accuracy": 0.1226928537624231,
      "eval_dim_256_euclidean_accuracy": 0.8797917652626597,
      "eval_dim_256_manhattan_accuracy": 0.8801467108376716,
      "eval_dim_256_max_accuracy": 0.8801467108376716,
      "eval_dim_384_cosine_accuracy": 0.879081874112636,
      "eval_dim_384_dot_accuracy": 0.12091812588736393,
      "eval_dim_384_euclidean_accuracy": 0.879081874112636,
      "eval_dim_384_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_384_max_accuracy": 0.8794368196876479,
      "eval_dim_64_cosine_accuracy": 0.8767155702792239,
      "eval_dim_64_dot_accuracy": 0.13002839564600094,
      "eval_dim_64_euclidean_accuracy": 0.8768338854708945,
      "eval_dim_64_manhattan_accuracy": 0.8778987221959299,
      "eval_dim_64_max_accuracy": 0.8778987221959299,
      "eval_loss": 16.423009872436523,
      "eval_runtime": 103.6087,
      "eval_samples_per_second": 81.576,
      "eval_sequential_score": 0.8767155702792239,
      "eval_steps_per_second": 2.558,
      "step": 2800
    },
    {
      "epoch": 19.183845183003786,
      "grad_norm": 6.176373481750488,
      "learning_rate": 1.5442433646315792e-05,
      "loss": 16.2052,
      "step": 2850
    },
    {
      "epoch": 19.183845183003786,
      "eval_dim_128_cosine_accuracy": 0.8769522006625651,
      "eval_dim_128_dot_accuracy": 0.12576904874585898,
      "eval_dim_128_euclidean_accuracy": 0.8758873639375295,
      "eval_dim_128_manhattan_accuracy": 0.8793185044959773,
      "eval_dim_128_max_accuracy": 0.8793185044959773,
      "eval_dim_256_cosine_accuracy": 0.8776620918125887,
      "eval_dim_256_dot_accuracy": 0.12328442972077615,
      "eval_dim_256_euclidean_accuracy": 0.8778987221959299,
      "eval_dim_256_manhattan_accuracy": 0.8796734500709891,
      "eval_dim_256_max_accuracy": 0.8796734500709891,
      "eval_dim_384_cosine_accuracy": 0.878490298154283,
      "eval_dim_384_dot_accuracy": 0.121509701845717,
      "eval_dim_384_euclidean_accuracy": 0.878490298154283,
      "eval_dim_384_manhattan_accuracy": 0.8814481779460482,
      "eval_dim_384_max_accuracy": 0.8814481779460482,
      "eval_dim_64_cosine_accuracy": 0.8744675816374823,
      "eval_dim_64_dot_accuracy": 0.13369616658778988,
      "eval_dim_64_euclidean_accuracy": 0.8742309512541411,
      "eval_dim_64_manhattan_accuracy": 0.8781353525792712,
      "eval_dim_64_max_accuracy": 0.8781353525792712,
      "eval_loss": 16.435117721557617,
      "eval_runtime": 104.4101,
      "eval_samples_per_second": 80.95,
      "eval_sequential_score": 0.8744675816374823,
      "eval_steps_per_second": 2.538,
      "step": 2850
    },
    {
      "epoch": 19.520403870424904,
      "grad_norm": 7.323819160461426,
      "learning_rate": 1.5243072835572319e-05,
      "loss": 16.186,
      "step": 2900
    },
    {
      "epoch": 19.520403870424904,
      "eval_dim_128_cosine_accuracy": 0.8776620918125887,
      "eval_dim_128_dot_accuracy": 0.12363937529578797,
      "eval_dim_128_euclidean_accuracy": 0.8776620918125887,
      "eval_dim_128_manhattan_accuracy": 0.876360624704212,
      "eval_dim_128_max_accuracy": 0.8776620918125887,
      "eval_dim_256_cosine_accuracy": 0.8793185044959773,
      "eval_dim_256_dot_accuracy": 0.12198296261239944,
      "eval_dim_256_euclidean_accuracy": 0.8789635589209654,
      "eval_dim_256_manhattan_accuracy": 0.8777804070042593,
      "eval_dim_256_max_accuracy": 0.8793185044959773,
      "eval_dim_384_cosine_accuracy": 0.8792001893043067,
      "eval_dim_384_dot_accuracy": 0.12079981069569333,
      "eval_dim_384_euclidean_accuracy": 0.8792001893043067,
      "eval_dim_384_manhattan_accuracy": 0.8789635589209654,
      "eval_dim_384_max_accuracy": 0.8792001893043067,
      "eval_dim_64_cosine_accuracy": 0.8762423095125415,
      "eval_dim_64_dot_accuracy": 0.13097491717936582,
      "eval_dim_64_euclidean_accuracy": 0.8748225272124941,
      "eval_dim_64_manhattan_accuracy": 0.8782536677709418,
      "eval_dim_64_max_accuracy": 0.8782536677709418,
      "eval_loss": 16.433080673217773,
      "eval_runtime": 101.0285,
      "eval_samples_per_second": 83.66,
      "eval_sequential_score": 0.8762423095125415,
      "eval_steps_per_second": 2.623,
      "step": 2900
    },
    {
      "epoch": 19.856962557846025,
      "grad_norm": 6.637113571166992,
      "learning_rate": 1.5040795557913246e-05,
      "loss": 16.1496,
      "step": 2950
    },
    {
      "epoch": 19.856962557846025,
      "eval_dim_128_cosine_accuracy": 0.8774254614292475,
      "eval_dim_128_dot_accuracy": 0.12529578797917654,
      "eval_dim_128_euclidean_accuracy": 0.8770705158542357,
      "eval_dim_128_manhattan_accuracy": 0.8782536677709418,
      "eval_dim_128_max_accuracy": 0.8782536677709418,
      "eval_dim_256_cosine_accuracy": 0.8781353525792712,
      "eval_dim_256_dot_accuracy": 0.12375769048745859,
      "eval_dim_256_euclidean_accuracy": 0.8783719829626124,
      "eval_dim_256_manhattan_accuracy": 0.879081874112636,
      "eval_dim_256_max_accuracy": 0.879081874112636,
      "eval_dim_384_cosine_accuracy": 0.8780170373876006,
      "eval_dim_384_dot_accuracy": 0.12198296261239944,
      "eval_dim_384_euclidean_accuracy": 0.8780170373876006,
      "eval_dim_384_manhattan_accuracy": 0.8786086133459536,
      "eval_dim_384_max_accuracy": 0.8786086133459536,
      "eval_dim_64_cosine_accuracy": 0.8770705158542357,
      "eval_dim_64_dot_accuracy": 0.13357785139611927,
      "eval_dim_64_euclidean_accuracy": 0.8756507335541883,
      "eval_dim_64_manhattan_accuracy": 0.8775437766209181,
      "eval_dim_64_max_accuracy": 0.8775437766209181,
      "eval_loss": 16.437721252441406,
      "eval_runtime": 103.9645,
      "eval_samples_per_second": 81.297,
      "eval_sequential_score": 0.8770705158542357,
      "eval_steps_per_second": 2.549,
      "step": 2950
    },
    {
      "epoch": 20.193521245267142,
      "grad_norm": 4.9336957931518555,
      "learning_rate": 1.4835714330369445e-05,
      "loss": 16.151,
      "step": 3000
    },
    {
      "epoch": 20.193521245267142,
      "eval_dim_128_cosine_accuracy": 0.8765972550875533,
      "eval_dim_128_dot_accuracy": 0.1261239943208708,
      "eval_dim_128_euclidean_accuracy": 0.8761239943208708,
      "eval_dim_128_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_128_max_accuracy": 0.8797917652626597,
      "eval_dim_256_cosine_accuracy": 0.8780170373876006,
      "eval_dim_256_dot_accuracy": 0.12245622337908188,
      "eval_dim_256_euclidean_accuracy": 0.8771888310459063,
      "eval_dim_256_manhattan_accuracy": 0.8801467108376716,
      "eval_dim_256_max_accuracy": 0.8801467108376716,
      "eval_dim_384_cosine_accuracy": 0.8780170373876006,
      "eval_dim_384_dot_accuracy": 0.12198296261239944,
      "eval_dim_384_euclidean_accuracy": 0.8780170373876006,
      "eval_dim_384_manhattan_accuracy": 0.8819214387127308,
      "eval_dim_384_max_accuracy": 0.8819214387127308,
      "eval_dim_64_cosine_accuracy": 0.8750591575958353,
      "eval_dim_64_dot_accuracy": 0.1361807856128727,
      "eval_dim_64_euclidean_accuracy": 0.8730477993374349,
      "eval_dim_64_manhattan_accuracy": 0.878490298154283,
      "eval_dim_64_max_accuracy": 0.878490298154283,
      "eval_loss": 16.44074821472168,
      "eval_runtime": 101.9564,
      "eval_samples_per_second": 82.898,
      "eval_sequential_score": 0.8750591575958353,
      "eval_steps_per_second": 2.599,
      "step": 3000
    },
    {
      "epoch": 20.530079932688263,
      "grad_norm": 5.225156784057617,
      "learning_rate": 1.4627943229672992e-05,
      "loss": 16.1081,
      "step": 3050
    },
    {
      "epoch": 20.530079932688263,
      "eval_dim_128_cosine_accuracy": 0.8758873639375295,
      "eval_dim_128_dot_accuracy": 0.1261239943208708,
      "eval_dim_128_euclidean_accuracy": 0.8758873639375295,
      "eval_dim_128_manhattan_accuracy": 0.8781353525792712,
      "eval_dim_128_max_accuracy": 0.8781353525792712,
      "eval_dim_256_cosine_accuracy": 0.8775437766209181,
      "eval_dim_256_dot_accuracy": 0.12245622337908188,
      "eval_dim_256_euclidean_accuracy": 0.8778987221959299,
      "eval_dim_256_manhattan_accuracy": 0.8776620918125887,
      "eval_dim_256_max_accuracy": 0.8778987221959299,
      "eval_dim_384_cosine_accuracy": 0.8774254614292475,
      "eval_dim_384_dot_accuracy": 0.12257453857075248,
      "eval_dim_384_euclidean_accuracy": 0.8774254614292475,
      "eval_dim_384_manhattan_accuracy": 0.8788452437292948,
      "eval_dim_384_max_accuracy": 0.8788452437292948,
      "eval_dim_64_cosine_accuracy": 0.8749408424041647,
      "eval_dim_64_dot_accuracy": 0.13712730714623758,
      "eval_dim_64_euclidean_accuracy": 0.8743492664458117,
      "eval_dim_64_manhattan_accuracy": 0.8765972550875533,
      "eval_dim_64_max_accuracy": 0.8765972550875533,
      "eval_loss": 16.442630767822266,
      "eval_runtime": 104.3455,
      "eval_samples_per_second": 81.0,
      "eval_sequential_score": 0.8749408424041647,
      "eval_steps_per_second": 2.54,
      "step": 3050
    },
    {
      "epoch": 20.86663862010938,
      "grad_norm": 4.5568132400512695,
      "learning_rate": 1.4417597828801833e-05,
      "loss": 16.0864,
      "step": 3100
    },
    {
      "epoch": 20.86663862010938,
      "eval_dim_128_cosine_accuracy": 0.8774254614292475,
      "eval_dim_128_dot_accuracy": 0.12659725508755323,
      "eval_dim_128_euclidean_accuracy": 0.8765972550875533,
      "eval_dim_128_manhattan_accuracy": 0.879081874112636,
      "eval_dim_128_max_accuracy": 0.879081874112636,
      "eval_dim_256_cosine_accuracy": 0.8781353525792712,
      "eval_dim_256_dot_accuracy": 0.12292948414576432,
      "eval_dim_256_euclidean_accuracy": 0.8780170373876006,
      "eval_dim_256_manhattan_accuracy": 0.880619971604354,
      "eval_dim_256_max_accuracy": 0.880619971604354,
      "eval_dim_384_cosine_accuracy": 0.8787269285376242,
      "eval_dim_384_dot_accuracy": 0.12127307146237577,
      "eval_dim_384_euclidean_accuracy": 0.8787269285376242,
      "eval_dim_384_manhattan_accuracy": 0.8793185044959773,
      "eval_dim_384_max_accuracy": 0.8793185044959773,
      "eval_dim_64_cosine_accuracy": 0.8745858968291529,
      "eval_dim_64_dot_accuracy": 0.13724562233790819,
      "eval_dim_64_euclidean_accuracy": 0.8744675816374823,
      "eval_dim_64_manhattan_accuracy": 0.8780170373876006,
      "eval_dim_64_max_accuracy": 0.8780170373876006,
      "eval_loss": 16.441152572631836,
      "eval_runtime": 103.8678,
      "eval_samples_per_second": 81.373,
      "eval_sequential_score": 0.8745858968291529,
      "eval_steps_per_second": 2.551,
      "step": 3100
    },
    {
      "epoch": 21.203197307530502,
      "grad_norm": 6.664557933807373,
      "learning_rate": 1.4204795132692146e-05,
      "loss": 16.0934,
      "step": 3150
    },
    {
      "epoch": 21.203197307530502,
      "eval_dim_128_cosine_accuracy": 0.8768338854708945,
      "eval_dim_128_dot_accuracy": 0.12789872219592996,
      "eval_dim_128_euclidean_accuracy": 0.8758873639375295,
      "eval_dim_128_manhattan_accuracy": 0.8803833412210128,
      "eval_dim_128_max_accuracy": 0.8803833412210128,
      "eval_dim_256_cosine_accuracy": 0.8782536677709418,
      "eval_dim_256_dot_accuracy": 0.12411263606247042,
      "eval_dim_256_euclidean_accuracy": 0.8777804070042593,
      "eval_dim_256_manhattan_accuracy": 0.88180312352106,
      "eval_dim_256_max_accuracy": 0.88180312352106,
      "eval_dim_384_cosine_accuracy": 0.8794368196876479,
      "eval_dim_384_dot_accuracy": 0.12056318031235211,
      "eval_dim_384_euclidean_accuracy": 0.8794368196876479,
      "eval_dim_384_manhattan_accuracy": 0.881211547562707,
      "eval_dim_384_max_accuracy": 0.881211547562707,
      "eval_dim_64_cosine_accuracy": 0.8745858968291529,
      "eval_dim_64_dot_accuracy": 0.14008518693800284,
      "eval_dim_64_euclidean_accuracy": 0.8729294841457643,
      "eval_dim_64_manhattan_accuracy": 0.8795551348793185,
      "eval_dim_64_max_accuracy": 0.8795551348793185,
      "eval_loss": 16.4547176361084,
      "eval_runtime": 105.011,
      "eval_samples_per_second": 80.487,
      "eval_sequential_score": 0.8745858968291529,
      "eval_steps_per_second": 2.524,
      "step": 3150
    },
    {
      "epoch": 21.53975599495162,
      "grad_norm": 6.669680118560791,
      "learning_rate": 1.3989653513154165e-05,
      "loss": 16.0382,
      "step": 3200
    },
    {
      "epoch": 21.53975599495162,
      "eval_dim_128_cosine_accuracy": 0.8742309512541411,
      "eval_dim_128_dot_accuracy": 0.1283719829626124,
      "eval_dim_128_euclidean_accuracy": 0.8738760056791292,
      "eval_dim_128_manhattan_accuracy": 0.8748225272124941,
      "eval_dim_128_max_accuracy": 0.8748225272124941,
      "eval_dim_256_cosine_accuracy": 0.8751774727875059,
      "eval_dim_256_dot_accuracy": 0.12446758163748226,
      "eval_dim_256_euclidean_accuracy": 0.8754141031708471,
      "eval_dim_256_manhattan_accuracy": 0.8765972550875533,
      "eval_dim_256_max_accuracy": 0.8765972550875533,
      "eval_dim_384_cosine_accuracy": 0.8765972550875533,
      "eval_dim_384_dot_accuracy": 0.12340274491244675,
      "eval_dim_384_euclidean_accuracy": 0.8765972550875533,
      "eval_dim_384_manhattan_accuracy": 0.8761239943208708,
      "eval_dim_384_max_accuracy": 0.8765972550875533,
      "eval_dim_64_cosine_accuracy": 0.8723379081874113,
      "eval_dim_64_dot_accuracy": 0.14020350212967345,
      "eval_dim_64_euclidean_accuracy": 0.8703265499290109,
      "eval_dim_64_manhattan_accuracy": 0.8754141031708471,
      "eval_dim_64_max_accuracy": 0.8754141031708471,
      "eval_loss": 16.458948135375977,
      "eval_runtime": 101.007,
      "eval_samples_per_second": 83.677,
      "eval_sequential_score": 0.8723379081874113,
      "eval_steps_per_second": 2.624,
      "step": 3200
    },
    {
      "epoch": 21.87631468237274,
      "grad_norm": 5.666304588317871,
      "learning_rate": 1.37722926430277e-05,
      "loss": 16.0279,
      "step": 3250
    },
    {
      "epoch": 21.87631468237274,
      "eval_dim_128_cosine_accuracy": 0.8751774727875059,
      "eval_dim_128_dot_accuracy": 0.12979176526265973,
      "eval_dim_128_euclidean_accuracy": 0.8743492664458117,
      "eval_dim_128_manhattan_accuracy": 0.8771888310459063,
      "eval_dim_128_max_accuracy": 0.8771888310459063,
      "eval_dim_256_cosine_accuracy": 0.8765972550875533,
      "eval_dim_256_dot_accuracy": 0.1260056791292002,
      "eval_dim_256_euclidean_accuracy": 0.8761239943208708,
      "eval_dim_256_manhattan_accuracy": 0.8796734500709891,
      "eval_dim_256_max_accuracy": 0.8796734500709891,
      "eval_dim_384_cosine_accuracy": 0.8773071462375769,
      "eval_dim_384_dot_accuracy": 0.1226928537624231,
      "eval_dim_384_euclidean_accuracy": 0.8773071462375769,
      "eval_dim_384_manhattan_accuracy": 0.8777804070042593,
      "eval_dim_384_max_accuracy": 0.8777804070042593,
      "eval_dim_64_cosine_accuracy": 0.8728111689540937,
      "eval_dim_64_dot_accuracy": 0.14221486038807382,
      "eval_dim_64_euclidean_accuracy": 0.8732844297207761,
      "eval_dim_64_manhattan_accuracy": 0.8776620918125887,
      "eval_dim_64_max_accuracy": 0.8776620918125887,
      "eval_loss": 16.46676254272461,
      "eval_runtime": 102.9103,
      "eval_samples_per_second": 82.13,
      "eval_sequential_score": 0.8728111689540937,
      "eval_steps_per_second": 2.575,
      "step": 3250
    },
    {
      "epoch": 22.212873369793858,
      "grad_norm": 6.600480556488037,
      "learning_rate": 1.3552833429613939e-05,
      "loss": 16.0327,
      "step": 3300
    },
    {
      "epoch": 22.212873369793858,
      "eval_dim_128_cosine_accuracy": 0.8742309512541411,
      "eval_dim_128_dot_accuracy": 0.13002839564600094,
      "eval_dim_128_euclidean_accuracy": 0.8742309512541411,
      "eval_dim_128_manhattan_accuracy": 0.880028395646001,
      "eval_dim_128_max_accuracy": 0.880028395646001,
      "eval_dim_256_cosine_accuracy": 0.8768338854708945,
      "eval_dim_256_dot_accuracy": 0.12363937529578797,
      "eval_dim_256_euclidean_accuracy": 0.8764789398958827,
      "eval_dim_256_manhattan_accuracy": 0.8814481779460482,
      "eval_dim_256_max_accuracy": 0.8814481779460482,
      "eval_dim_384_cosine_accuracy": 0.8773071462375769,
      "eval_dim_384_dot_accuracy": 0.1226928537624231,
      "eval_dim_384_euclidean_accuracy": 0.8773071462375769,
      "eval_dim_384_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_384_max_accuracy": 0.8807382867960246,
      "eval_dim_64_cosine_accuracy": 0.8726928537624231,
      "eval_dim_64_dot_accuracy": 0.1432796971131093,
      "eval_dim_64_euclidean_accuracy": 0.869971604353999,
      "eval_dim_64_manhattan_accuracy": 0.8795551348793185,
      "eval_dim_64_max_accuracy": 0.8795551348793185,
      "eval_loss": 16.47365379333496,
      "eval_runtime": 104.4255,
      "eval_samples_per_second": 80.938,
      "eval_sequential_score": 0.8726928537624231,
      "eval_steps_per_second": 2.538,
      "step": 3300
    },
    {
      "epoch": 22.549432057214975,
      "grad_norm": 7.925108432769775,
      "learning_rate": 1.3331397947420578e-05,
      "loss": 15.979,
      "step": 3350
    },
    {
      "epoch": 22.549432057214975,
      "eval_dim_128_cosine_accuracy": 0.8739943208707998,
      "eval_dim_128_dot_accuracy": 0.1293185044959773,
      "eval_dim_128_euclidean_accuracy": 0.8732844297207761,
      "eval_dim_128_manhattan_accuracy": 0.8782536677709418,
      "eval_dim_128_max_accuracy": 0.8782536677709418,
      "eval_dim_256_cosine_accuracy": 0.8770705158542357,
      "eval_dim_256_dot_accuracy": 0.12328442972077615,
      "eval_dim_256_euclidean_accuracy": 0.8773071462375769,
      "eval_dim_256_manhattan_accuracy": 0.8793185044959773,
      "eval_dim_256_max_accuracy": 0.8793185044959773,
      "eval_dim_384_cosine_accuracy": 0.8770705158542357,
      "eval_dim_384_dot_accuracy": 0.12292948414576432,
      "eval_dim_384_euclidean_accuracy": 0.8770705158542357,
      "eval_dim_384_manhattan_accuracy": 0.8778987221959299,
      "eval_dim_384_max_accuracy": 0.8778987221959299,
      "eval_dim_64_cosine_accuracy": 0.8722195929957407,
      "eval_dim_64_dot_accuracy": 0.14162328442972077,
      "eval_dim_64_euclidean_accuracy": 0.8700899195456696,
      "eval_dim_64_manhattan_accuracy": 0.8767155702792239,
      "eval_dim_64_max_accuracy": 0.8767155702792239,
      "eval_loss": 16.468605041503906,
      "eval_runtime": 101.8518,
      "eval_samples_per_second": 82.983,
      "eval_sequential_score": 0.8722195929957407,
      "eval_steps_per_second": 2.602,
      "step": 3350
    },
    {
      "epoch": 22.885990744636096,
      "grad_norm": 6.396854877471924,
      "learning_rate": 1.3108109370257714e-05,
      "loss": 15.9622,
      "step": 3400
    },
    {
      "epoch": 22.885990744636096,
      "eval_dim_128_cosine_accuracy": 0.8743492664458117,
      "eval_dim_128_dot_accuracy": 0.13002839564600094,
      "eval_dim_128_euclidean_accuracy": 0.873639375295788,
      "eval_dim_128_manhattan_accuracy": 0.8786086133459536,
      "eval_dim_128_max_accuracy": 0.8786086133459536,
      "eval_dim_256_cosine_accuracy": 0.8760056791292002,
      "eval_dim_256_dot_accuracy": 0.12434926644581164,
      "eval_dim_256_euclidean_accuracy": 0.8757690487458589,
      "eval_dim_256_manhattan_accuracy": 0.8805016564126834,
      "eval_dim_256_max_accuracy": 0.8805016564126834,
      "eval_dim_384_cosine_accuracy": 0.8764789398958827,
      "eval_dim_384_dot_accuracy": 0.12352106010411737,
      "eval_dim_384_euclidean_accuracy": 0.8764789398958827,
      "eval_dim_384_manhattan_accuracy": 0.8807382867960246,
      "eval_dim_384_max_accuracy": 0.8807382867960246,
      "eval_dim_64_cosine_accuracy": 0.8721012778040701,
      "eval_dim_64_dot_accuracy": 0.14351632749645055,
      "eval_dim_64_euclidean_accuracy": 0.8703265499290109,
      "eval_dim_64_manhattan_accuracy": 0.8781353525792712,
      "eval_dim_64_max_accuracy": 0.8781353525792712,
      "eval_loss": 16.473587036132812,
      "eval_runtime": 103.4538,
      "eval_samples_per_second": 81.698,
      "eval_sequential_score": 0.8721012778040701,
      "eval_steps_per_second": 2.562,
      "step": 3400
    },
    {
      "epoch": 23.222549432057214,
      "grad_norm": 4.757622241973877,
      "learning_rate": 1.288309190272222e-05,
      "loss": 15.9881,
      "step": 3450
    },
    {
      "epoch": 23.222549432057214,
      "eval_dim_128_cosine_accuracy": 0.8743492664458117,
      "eval_dim_128_dot_accuracy": 0.13097491717936582,
      "eval_dim_128_euclidean_accuracy": 0.8737576904874585,
      "eval_dim_128_manhattan_accuracy": 0.879081874112636,
      "eval_dim_128_max_accuracy": 0.879081874112636,
      "eval_dim_256_cosine_accuracy": 0.8756507335541883,
      "eval_dim_256_dot_accuracy": 0.12588736393752958,
      "eval_dim_256_euclidean_accuracy": 0.8747042120208235,
      "eval_dim_256_manhattan_accuracy": 0.8795551348793185,
      "eval_dim_256_max_accuracy": 0.8795551348793185,
      "eval_dim_384_cosine_accuracy": 0.8755324183625177,
      "eval_dim_384_dot_accuracy": 0.12446758163748226,
      "eval_dim_384_euclidean_accuracy": 0.8755324183625177,
      "eval_dim_384_manhattan_accuracy": 0.879081874112636,
      "eval_dim_384_max_accuracy": 0.879081874112636,
      "eval_dim_64_cosine_accuracy": 0.8723379081874113,
      "eval_dim_64_dot_accuracy": 0.14375295787979175,
      "eval_dim_64_euclidean_accuracy": 0.8700899195456696,
      "eval_dim_64_manhattan_accuracy": 0.8788452437292948,
      "eval_dim_64_max_accuracy": 0.8788452437292948,
      "eval_loss": 16.48019790649414,
      "eval_runtime": 104.0826,
      "eval_samples_per_second": 81.205,
      "eval_sequential_score": 0.8723379081874113,
      "eval_steps_per_second": 2.546,
      "step": 3450
    },
    {
      "epoch": 23.559108119478335,
      "grad_norm": 5.279081344604492,
      "learning_rate": 1.2656470711108763e-05,
      "loss": 15.9482,
      "step": 3500
    },
    {
      "epoch": 23.559108119478335,
      "eval_dim_128_cosine_accuracy": 0.8724562233790819,
      "eval_dim_128_dot_accuracy": 0.13073828679602462,
      "eval_dim_128_euclidean_accuracy": 0.8728111689540937,
      "eval_dim_128_manhattan_accuracy": 0.8783719829626124,
      "eval_dim_128_max_accuracy": 0.8783719829626124,
      "eval_dim_256_cosine_accuracy": 0.8761239943208708,
      "eval_dim_256_dot_accuracy": 0.1250591575958353,
      "eval_dim_256_euclidean_accuracy": 0.8761239943208708,
      "eval_dim_256_manhattan_accuracy": 0.8797917652626597,
      "eval_dim_256_max_accuracy": 0.8797917652626597,
      "eval_dim_384_cosine_accuracy": 0.8761239943208708,
      "eval_dim_384_dot_accuracy": 0.1238760056791292,
      "eval_dim_384_euclidean_accuracy": 0.8761239943208708,
      "eval_dim_384_manhattan_accuracy": 0.8770705158542357,
      "eval_dim_384_max_accuracy": 0.8770705158542357,
      "eval_dim_64_cosine_accuracy": 0.8710364410790346,
      "eval_dim_64_dot_accuracy": 0.143989588263133,
      "eval_dim_64_euclidean_accuracy": 0.867841930903928,
      "eval_dim_64_manhattan_accuracy": 0.8764789398958827,
      "eval_dim_64_max_accuracy": 0.8764789398958827,
      "eval_loss": 16.482074737548828,
      "eval_runtime": 102.3602,
      "eval_samples_per_second": 82.571,
      "eval_sequential_score": 0.8710364410790346,
      "eval_steps_per_second": 2.589,
      "step": 3500
    },
    {
      "epoch": 23.895666806899452,
      "grad_norm": 5.999639511108398,
      "learning_rate": 1.2428371853785872e-05,
      "loss": 15.9228,
      "step": 3550
    },
    {
      "epoch": 23.895666806899452,
      "eval_dim_128_cosine_accuracy": 0.8725745385707525,
      "eval_dim_128_dot_accuracy": 0.13310459062943683,
      "eval_dim_128_euclidean_accuracy": 0.8719829626123994,
      "eval_dim_128_manhattan_accuracy": 0.878490298154283,
      "eval_dim_128_max_accuracy": 0.878490298154283,
      "eval_dim_256_cosine_accuracy": 0.8748225272124941,
      "eval_dim_256_dot_accuracy": 0.12671557027922387,
      "eval_dim_256_euclidean_accuracy": 0.8743492664458117,
      "eval_dim_256_manhattan_accuracy": 0.879081874112636,
      "eval_dim_256_max_accuracy": 0.879081874112636,
      "eval_dim_384_cosine_accuracy": 0.8750591575958353,
      "eval_dim_384_dot_accuracy": 0.1249408424041647,
      "eval_dim_384_euclidean_accuracy": 0.8750591575958353,
      "eval_dim_384_manhattan_accuracy": 0.8781353525792712,
      "eval_dim_384_max_accuracy": 0.8781353525792712,
      "eval_dim_64_cosine_accuracy": 0.870918125887364,
      "eval_dim_64_dot_accuracy": 0.14469947941315664,
      "eval_dim_64_euclidean_accuracy": 0.8691433980123048,
      "eval_dim_64_manhattan_accuracy": 0.8776620918125887,
      "eval_dim_64_max_accuracy": 0.8776620918125887,
      "eval_loss": 16.499635696411133,
      "eval_runtime": 103.3405,
      "eval_samples_per_second": 81.788,
      "eval_sequential_score": 0.870918125887364,
      "eval_steps_per_second": 2.564,
      "step": 3550
    },
    {
      "epoch": 24.232225494320573,
      "grad_norm": 6.511181354522705,
      "learning_rate": 1.2198922211075779e-05,
      "loss": 15.9418,
      "step": 3600
    },
    {
      "epoch": 24.232225494320573,
      "eval_dim_128_cosine_accuracy": 0.870918125887364,
      "eval_dim_128_dot_accuracy": 0.1353525792711784,
      "eval_dim_128_euclidean_accuracy": 0.8703265499290109,
      "eval_dim_128_manhattan_accuracy": 0.8783719829626124,
      "eval_dim_128_max_accuracy": 0.8783719829626124,
      "eval_dim_256_cosine_accuracy": 0.8728111689540937,
      "eval_dim_256_dot_accuracy": 0.12884524372929484,
      "eval_dim_256_euclidean_accuracy": 0.8721012778040701,
      "eval_dim_256_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_256_max_accuracy": 0.8794368196876479,
      "eval_dim_384_cosine_accuracy": 0.8734027449124467,
      "eval_dim_384_dot_accuracy": 0.12659725508755323,
      "eval_dim_384_euclidean_accuracy": 0.8734027449124467,
      "eval_dim_384_manhattan_accuracy": 0.8794368196876479,
      "eval_dim_384_max_accuracy": 0.8794368196876479,
      "eval_dim_64_cosine_accuracy": 0.8698532891623284,
      "eval_dim_64_dot_accuracy": 0.14564600094652153,
      "eval_dim_64_euclidean_accuracy": 0.8680785612872692,
      "eval_dim_64_manhattan_accuracy": 0.8770705158542357,
      "eval_dim_64_max_accuracy": 0.8770705158542357,
      "eval_loss": 16.497343063354492,
      "eval_runtime": 104.6868,
      "eval_samples_per_second": 80.736,
      "eval_sequential_score": 0.8698532891623284,
      "eval_steps_per_second": 2.531,
      "step": 3600
    },
    {
      "epoch": 24.56878418174169,
      "grad_norm": 5.682207107543945,
      "learning_rate": 1.1968249414677055e-05,
      "loss": 15.896,
      "step": 3650
    },
    {
      "epoch": 24.56878418174169,
      "eval_dim_128_cosine_accuracy": 0.8696166587789872,
      "eval_dim_128_dot_accuracy": 0.13487931850449597,
      "eval_dim_128_euclidean_accuracy": 0.8685518220539518,
      "eval_dim_128_manhattan_accuracy": 0.8764789398958827,
      "eval_dim_128_max_accuracy": 0.8764789398958827,
      "eval_dim_256_cosine_accuracy": 0.8716280170373876,
      "eval_dim_256_dot_accuracy": 0.12896355892096545,
      "eval_dim_256_euclidean_accuracy": 0.871509701845717,
      "eval_dim_256_manhattan_accuracy": 0.8777804070042593,
      "eval_dim_256_max_accuracy": 0.8777804070042593,
      "eval_dim_384_cosine_accuracy": 0.8726928537624231,
      "eval_dim_384_dot_accuracy": 0.1273071462375769,
      "eval_dim_384_euclidean_accuracy": 0.8726928537624231,
      "eval_dim_384_manhattan_accuracy": 0.8773071462375769,
      "eval_dim_384_max_accuracy": 0.8773071462375769,
      "eval_dim_64_cosine_accuracy": 0.8685518220539518,
      "eval_dim_64_dot_accuracy": 0.14694746805489825,
      "eval_dim_64_euclidean_accuracy": 0.8659488878371983,
      "eval_dim_64_manhattan_accuracy": 0.8760056791292002,
      "eval_dim_64_max_accuracy": 0.8760056791292002,
      "eval_loss": 16.498498916625977,
      "eval_runtime": 102.6029,
      "eval_samples_per_second": 82.376,
      "eval_sequential_score": 0.8685518220539518,
      "eval_steps_per_second": 2.583,
      "step": 3650
    },
    {
      "epoch": 24.90534286916281,
      "grad_norm": 5.5915117263793945,
      "learning_rate": 1.1736481776669307e-05,
      "loss": 15.8788,
      "step": 3700
    },
    {
      "epoch": 24.90534286916281,
      "eval_dim_128_cosine_accuracy": 0.8691433980123048,
      "eval_dim_128_dot_accuracy": 0.1361807856128727,
      "eval_dim_128_euclidean_accuracy": 0.8697349739706578,
      "eval_dim_128_manhattan_accuracy": 0.8747042120208235,
      "eval_dim_128_max_accuracy": 0.8747042120208235,
      "eval_dim_256_cosine_accuracy": 0.871509701845717,
      "eval_dim_256_dot_accuracy": 0.13073828679602462,
      "eval_dim_256_euclidean_accuracy": 0.8704448651206815,
      "eval_dim_256_manhattan_accuracy": 0.8770705158542357,
      "eval_dim_256_max_accuracy": 0.8770705158542357,
      "eval_dim_384_cosine_accuracy": 0.8717463322290582,
      "eval_dim_384_dot_accuracy": 0.1282536677709418,
      "eval_dim_384_euclidean_accuracy": 0.8717463322290582,
      "eval_dim_384_manhattan_accuracy": 0.8758873639375295,
      "eval_dim_384_max_accuracy": 0.8758873639375295,
      "eval_dim_64_cosine_accuracy": 0.8661855182205395,
      "eval_dim_64_dot_accuracy": 0.14824893516327498,
      "eval_dim_64_euclidean_accuracy": 0.8667770941788926,
      "eval_dim_64_manhattan_accuracy": 0.8744675816374823,
      "eval_dim_64_max_accuracy": 0.8744675816374823,
      "eval_loss": 16.517175674438477,
      "eval_runtime": 103.5179,
      "eval_samples_per_second": 81.648,
      "eval_sequential_score": 0.8661855182205395,
      "eval_steps_per_second": 2.56,
      "step": 3700
    },
    {
      "epoch": 25.24190155658393,
      "grad_norm": 5.408066749572754,
      "learning_rate": 1.150374821813937e-05,
      "loss": 15.9147,
      "step": 3750
    },
    {
      "epoch": 25.24190155658393,
      "eval_dim_128_cosine_accuracy": 0.8677236157122574,
      "eval_dim_128_dot_accuracy": 0.13724562233790819,
      "eval_dim_128_euclidean_accuracy": 0.8673686701372456,
      "eval_dim_128_manhattan_accuracy": 0.8768338854708945,
      "eval_dim_128_max_accuracy": 0.8768338854708945,
      "eval_dim_256_cosine_accuracy": 0.8705631803123521,
      "eval_dim_256_dot_accuracy": 0.13144817794604827,
      "eval_dim_256_euclidean_accuracy": 0.869971604353999,
      "eval_dim_256_manhattan_accuracy": 0.8782536677709418,
      "eval_dim_256_max_accuracy": 0.8782536677709418,
      "eval_dim_384_cosine_accuracy": 0.8711547562707052,
      "eval_dim_384_dot_accuracy": 0.12884524372929484,
      "eval_dim_384_euclidean_accuracy": 0.8711547562707052,
      "eval_dim_384_manhattan_accuracy": 0.8782536677709418,
      "eval_dim_384_max_accuracy": 0.8782536677709418,
      "eval_dim_64_cosine_accuracy": 0.8661855182205395,
      "eval_dim_64_dot_accuracy": 0.14955040227165167,
      "eval_dim_64_euclidean_accuracy": 0.866658778987222,
      "eval_dim_64_manhattan_accuracy": 0.8768338854708945,
      "eval_dim_64_max_accuracy": 0.8768338854708945,
      "eval_loss": 16.506189346313477,
      "eval_runtime": 103.9114,
      "eval_samples_per_second": 81.339,
      "eval_sequential_score": 0.8661855182205395,
      "eval_steps_per_second": 2.55,
      "step": 3750
    },
    {
      "epoch": 25.578460244005047,
      "grad_norm": 6.964442253112793,
      "learning_rate": 1.1270178197468788e-05,
      "loss": 15.857,
      "step": 3800
    },
    {
      "epoch": 25.578460244005047,
      "eval_dim_128_cosine_accuracy": 0.8683151916706106,
      "eval_dim_128_dot_accuracy": 0.13712730714623758,
      "eval_dim_128_euclidean_accuracy": 0.86819687647894,
      "eval_dim_128_manhattan_accuracy": 0.8739943208707998,
      "eval_dim_128_max_accuracy": 0.8739943208707998,
      "eval_dim_256_cosine_accuracy": 0.8717463322290582,
      "eval_dim_256_dot_accuracy": 0.13073828679602462,
      "eval_dim_256_euclidean_accuracy": 0.871509701845717,
      "eval_dim_256_manhattan_accuracy": 0.8761239943208708,
      "eval_dim_256_max_accuracy": 0.8761239943208708,
      "eval_dim_384_cosine_accuracy": 0.8731661145291055,
      "eval_dim_384_dot_accuracy": 0.12683388547089447,
      "eval_dim_384_euclidean_accuracy": 0.8731661145291055,
      "eval_dim_384_manhattan_accuracy": 0.8755324183625177,
      "eval_dim_384_max_accuracy": 0.8755324183625177,
      "eval_dim_64_cosine_accuracy": 0.8663038334122102,
      "eval_dim_64_dot_accuracy": 0.1499053478466635,
      "eval_dim_64_euclidean_accuracy": 0.865120681495504,
      "eval_dim_64_manhattan_accuracy": 0.8748225272124941,
      "eval_dim_64_max_accuracy": 0.8748225272124941,
      "eval_loss": 16.505783081054688,
      "eval_runtime": 102.7207,
      "eval_samples_per_second": 82.281,
      "eval_sequential_score": 0.8663038334122102,
      "eval_steps_per_second": 2.58,
      "step": 3800
    },
    {
      "epoch": 25.915018931426168,
      "grad_norm": 17.978727340698242,
      "learning_rate": 1.1035901638322392e-05,
      "loss": 15.8291,
      "step": 3850
    },
    {
      "epoch": 25.915018931426168,
      "eval_dim_128_cosine_accuracy": 0.8673686701372456,
      "eval_dim_128_dot_accuracy": 0.13771888310459063,
      "eval_dim_128_euclidean_accuracy": 0.8665404637955514,
      "eval_dim_128_manhattan_accuracy": 0.8745858968291529,
      "eval_dim_128_max_accuracy": 0.8745858968291529,
      "eval_dim_256_cosine_accuracy": 0.8702082347373403,
      "eval_dim_256_dot_accuracy": 0.1320397539044013,
      "eval_dim_256_euclidean_accuracy": 0.868788452437293,
      "eval_dim_256_manhattan_accuracy": 0.8756507335541883,
      "eval_dim_256_max_accuracy": 0.8756507335541883,
      "eval_dim_384_cosine_accuracy": 0.8705631803123521,
      "eval_dim_384_dot_accuracy": 0.1294368196876479,
      "eval_dim_384_euclidean_accuracy": 0.8705631803123521,
      "eval_dim_384_manhattan_accuracy": 0.8762423095125415,
      "eval_dim_384_max_accuracy": 0.8762423095125415,
      "eval_dim_64_cosine_accuracy": 0.8644107903454804,
      "eval_dim_64_dot_accuracy": 0.15309985802177,
      "eval_dim_64_euclidean_accuracy": 0.8640558447704685,
      "eval_dim_64_manhattan_accuracy": 0.8731661145291055,
      "eval_dim_64_max_accuracy": 0.8731661145291055,
      "eval_loss": 16.520679473876953,
      "eval_runtime": 104.1552,
      "eval_samples_per_second": 81.148,
      "eval_sequential_score": 0.8644107903454804,
      "eval_steps_per_second": 2.544,
      "step": 3850
    },
    {
      "epoch": 26.251577618847286,
      "grad_norm": 7.759204387664795,
      "learning_rate": 1.080104885737807e-05,
      "loss": 15.8802,
      "step": 3900
    },
    {
      "epoch": 26.251577618847286,
      "eval_dim_128_cosine_accuracy": 0.867841930903928,
      "eval_dim_128_dot_accuracy": 0.13913866540463796,
      "eval_dim_128_euclidean_accuracy": 0.86819687647894,
      "eval_dim_128_manhattan_accuracy": 0.8750591575958353,
      "eval_dim_128_max_accuracy": 0.8750591575958353,
      "eval_dim_256_cosine_accuracy": 0.8697349739706578,
      "eval_dim_256_dot_accuracy": 0.1318031235210601,
      "eval_dim_256_euclidean_accuracy": 0.8697349739706578,
      "eval_dim_256_manhattan_accuracy": 0.8764789398958827,
      "eval_dim_256_max_accuracy": 0.8764789398958827,
      "eval_dim_384_cosine_accuracy": 0.8713913866540464,
      "eval_dim_384_dot_accuracy": 0.1286086133459536,
      "eval_dim_384_euclidean_accuracy": 0.8713913866540464,
      "eval_dim_384_manhattan_accuracy": 0.8762423095125415,
      "eval_dim_384_max_accuracy": 0.8762423095125415,
      "eval_dim_64_cosine_accuracy": 0.8664221486038808,
      "eval_dim_64_dot_accuracy": 0.15061523899668716,
      "eval_dim_64_euclidean_accuracy": 0.8655939422621864,
      "eval_dim_64_manhattan_accuracy": 0.8737576904874585,
      "eval_dim_64_max_accuracy": 0.8737576904874585,
      "eval_loss": 16.52326011657715,
      "eval_runtime": 103.0214,
      "eval_samples_per_second": 82.041,
      "eval_sequential_score": 0.8664221486038808,
      "eval_steps_per_second": 2.572,
      "step": 3900
    },
    {
      "epoch": 26.588136306268407,
      "grad_norm": 6.951057434082031,
      "learning_rate": 1.0565750491837925e-05,
      "loss": 15.846,
      "step": 3950
    },
    {
      "epoch": 26.588136306268407,
      "eval_dim_128_cosine_accuracy": 0.8685518220539518,
      "eval_dim_128_dot_accuracy": 0.13689067676289635,
      "eval_dim_128_euclidean_accuracy": 0.867841930903928,
      "eval_dim_128_manhattan_accuracy": 0.8729294841457643,
      "eval_dim_128_max_accuracy": 0.8729294841457643,
      "eval_dim_256_cosine_accuracy": 0.8712730714623758,
      "eval_dim_256_dot_accuracy": 0.13097491717936582,
      "eval_dim_256_euclidean_accuracy": 0.8704448651206815,
      "eval_dim_256_manhattan_accuracy": 0.8765972550875533,
      "eval_dim_256_max_accuracy": 0.8765972550875533,
      "eval_dim_384_cosine_accuracy": 0.8717463322290582,
      "eval_dim_384_dot_accuracy": 0.1282536677709418,
      "eval_dim_384_euclidean_accuracy": 0.8717463322290582,
      "eval_dim_384_manhattan_accuracy": 0.8741126360624705,
      "eval_dim_384_max_accuracy": 0.8741126360624705,
      "eval_dim_64_cosine_accuracy": 0.8654756270705158,
      "eval_dim_64_dot_accuracy": 0.1499053478466635,
      "eval_dim_64_euclidean_accuracy": 0.865120681495504,
      "eval_dim_64_manhattan_accuracy": 0.8729294841457643,
      "eval_dim_64_max_accuracy": 0.8729294841457643,
      "eval_loss": 16.517038345336914,
      "eval_runtime": 103.0824,
      "eval_samples_per_second": 81.993,
      "eval_sequential_score": 0.8654756270705158,
      "eval_steps_per_second": 2.571,
      "step": 3950
    },
    {
      "epoch": 26.924694993689524,
      "grad_norm": 7.398913860321045,
      "learning_rate": 1.0330137426761136e-05,
      "loss": 15.8012,
      "step": 4000
    },
    {
      "epoch": 26.924694993689524,
      "eval_dim_128_cosine_accuracy": 0.8663038334122102,
      "eval_dim_128_dot_accuracy": 0.1386654046379555,
      "eval_dim_128_euclidean_accuracy": 0.865712257453857,
      "eval_dim_128_manhattan_accuracy": 0.8742309512541411,
      "eval_dim_128_max_accuracy": 0.8742309512541411,
      "eval_dim_256_cosine_accuracy": 0.86819687647894,
      "eval_dim_256_dot_accuracy": 0.1319214387127307,
      "eval_dim_256_euclidean_accuracy": 0.8680785612872692,
      "eval_dim_256_manhattan_accuracy": 0.8762423095125415,
      "eval_dim_256_max_accuracy": 0.8762423095125415,
      "eval_dim_384_cosine_accuracy": 0.8698532891623284,
      "eval_dim_384_dot_accuracy": 0.13014671083767157,
      "eval_dim_384_euclidean_accuracy": 0.8698532891623284,
      "eval_dim_384_manhattan_accuracy": 0.8737576904874585,
      "eval_dim_384_max_accuracy": 0.8737576904874585,
      "eval_dim_64_cosine_accuracy": 0.8634642688121155,
      "eval_dim_64_dot_accuracy": 0.1508518693800284,
      "eval_dim_64_euclidean_accuracy": 0.8637008991954567,
      "eval_dim_64_manhattan_accuracy": 0.8728111689540937,
      "eval_dim_64_max_accuracy": 0.8728111689540937,
      "eval_loss": 16.53356170654297,
      "eval_runtime": 104.1737,
      "eval_samples_per_second": 81.134,
      "eval_sequential_score": 0.8634642688121155,
      "eval_steps_per_second": 2.544,
      "step": 4000
    }
  ],
  "logging_steps": 50,
  "max_steps": 7400,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 50,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}