{
  "best_metric": 0.513288052386553,
  "best_model_checkpoint": "bkai-fine-tuned-legal/checkpoint-240",
  "epoch": 4.0,
  "eval_steps": 500,
  "global_step": 240,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "grad_norm": 3051630.25,
      "learning_rate": 2.9970400926424075e-05,
      "loss": 0.9663,
      "step": 60
    },
    {
      "epoch": 1.0,
      "eval_dim_128_cosine_accuracy@1": 0.2080602426439827,
      "eval_dim_128_cosine_accuracy@10": 0.6729884256031237,
      "eval_dim_128_cosine_accuracy@3": 0.4049644401059824,
      "eval_dim_128_cosine_accuracy@5": 0.49504950495049505,
      "eval_dim_128_cosine_map@100": 0.34660918730461937,
      "eval_dim_128_cosine_mrr@10": 0.3358910891089097,
      "eval_dim_128_cosine_ndcg@10": 0.41500734332231626,
      "eval_dim_128_cosine_precision@1": 0.2080602426439827,
      "eval_dim_128_cosine_precision@10": 0.06729884256031238,
      "eval_dim_128_cosine_precision@3": 0.13498814670199413,
      "eval_dim_128_cosine_precision@5": 0.09900990099009901,
      "eval_dim_128_cosine_recall@1": 0.2080602426439827,
      "eval_dim_128_cosine_recall@10": 0.6729884256031237,
      "eval_dim_128_cosine_recall@3": 0.4049644401059824,
      "eval_dim_128_cosine_recall@5": 0.49504950495049505,
      "eval_dim_256_cosine_accuracy@1": 0.21321991354064984,
      "eval_dim_256_cosine_accuracy@10": 0.6862362292567284,
      "eval_dim_256_cosine_accuracy@3": 0.41598103472319065,
      "eval_dim_256_cosine_accuracy@5": 0.5063450006972529,
      "eval_dim_256_cosine_map@100": 0.3546268888431031,
      "eval_dim_256_cosine_mrr@10": 0.34420854057236694,
      "eval_dim_256_cosine_ndcg@10": 0.424598027833536,
      "eval_dim_256_cosine_precision@1": 0.21321991354064984,
      "eval_dim_256_cosine_precision@10": 0.06862362292567285,
      "eval_dim_256_cosine_precision@3": 0.1386603449077302,
      "eval_dim_256_cosine_precision@5": 0.10126900013945056,
      "eval_dim_256_cosine_recall@1": 0.21321991354064984,
      "eval_dim_256_cosine_recall@10": 0.6862362292567284,
      "eval_dim_256_cosine_recall@3": 0.41598103472319065,
      "eval_dim_256_cosine_recall@5": 0.5063450006972529,
      "eval_dim_512_cosine_accuracy@1": 0.21433551805884815,
      "eval_dim_512_cosine_accuracy@10": 0.6919537024124948,
      "eval_dim_512_cosine_accuracy@3": 0.4211407056198578,
      "eval_dim_512_cosine_accuracy@5": 0.508855110863199,
      "eval_dim_512_cosine_map@100": 0.3564257538175301,
      "eval_dim_512_cosine_mrr@10": 0.34621192501543907,
      "eval_dim_512_cosine_ndcg@10": 0.42739973393443165,
      "eval_dim_512_cosine_precision@1": 0.21433551805884815,
      "eval_dim_512_cosine_precision@10": 0.06919537024124948,
      "eval_dim_512_cosine_precision@3": 0.14038023520661924,
      "eval_dim_512_cosine_precision@5": 0.1017710221726398,
      "eval_dim_512_cosine_recall@1": 0.21433551805884815,
      "eval_dim_512_cosine_recall@10": 0.6919537024124948,
      "eval_dim_512_cosine_recall@3": 0.4211407056198578,
      "eval_dim_512_cosine_recall@5": 0.508855110863199,
      "eval_dim_64_cosine_accuracy@1": 0.19704364802677451,
      "eval_dim_64_cosine_accuracy@10": 0.6503974341096082,
      "eval_dim_64_cosine_accuracy@3": 0.3846046576488635,
      "eval_dim_64_cosine_accuracy@5": 0.48096499790824154,
      "eval_dim_64_cosine_map@100": 0.3332997375022508,
      "eval_dim_64_cosine_mrr@10": 0.3223062799237654,
      "eval_dim_64_cosine_ndcg@10": 0.399385228043632,
      "eval_dim_64_cosine_precision@1": 0.19704364802677451,
      "eval_dim_64_cosine_precision@10": 0.06503974341096082,
      "eval_dim_64_cosine_precision@3": 0.12820155254962115,
      "eval_dim_64_cosine_precision@5": 0.0961929995816483,
      "eval_dim_64_cosine_recall@1": 0.19704364802677451,
      "eval_dim_64_cosine_recall@10": 0.6503974341096082,
      "eval_dim_64_cosine_recall@3": 0.3846046576488635,
      "eval_dim_64_cosine_recall@5": 0.48096499790824154,
      "eval_dim_768_cosine_accuracy@1": 0.22116859573281272,
      "eval_dim_768_cosine_accuracy@10": 0.6947427137079906,
      "eval_dim_768_cosine_accuracy@3": 0.4196067494073351,
      "eval_dim_768_cosine_accuracy@5": 0.5128991772416678,
      "eval_dim_768_cosine_map@100": 0.3609325312259912,
      "eval_dim_768_cosine_mrr@10": 0.3508835189353944,
      "eval_dim_768_cosine_ndcg@10": 0.4315716703139609,
      "eval_dim_768_cosine_precision@1": 0.22116859573281272,
      "eval_dim_768_cosine_precision@10": 0.06947427137079905,
      "eval_dim_768_cosine_precision@3": 0.13986891646911168,
      "eval_dim_768_cosine_precision@5": 0.10257983544833356,
      "eval_dim_768_cosine_recall@1": 0.22116859573281272,
      "eval_dim_768_cosine_recall@10": 0.6947427137079906,
      "eval_dim_768_cosine_recall@3": 0.4196067494073351,
      "eval_dim_768_cosine_recall@5": 0.5128991772416678,
      "eval_loss": 0.30127617716789246,
      "eval_runtime": 4760.4689,
      "eval_samples_per_second": 1.507,
      "eval_sequential_score": 0.399385228043632,
      "eval_steps_per_second": 0.019,
      "step": 60
    },
    {
      "epoch": 2.0,
      "grad_norm": 1582530.375,
      "learning_rate": 2.655769864163684e-05,
      "loss": 0.2399,
      "step": 120
    },
    {
      "epoch": 2.0,
      "eval_dim_128_cosine_accuracy@1": 0.3225491563240831,
      "eval_dim_128_cosine_accuracy@10": 0.7071538139729466,
      "eval_dim_128_cosine_accuracy@3": 0.46102356714544696,
      "eval_dim_128_cosine_accuracy@5": 0.53772137777158,
      "eval_dim_128_cosine_map@100": 0.43211454127792864,
      "eval_dim_128_cosine_mrr@10": 0.42227551225947657,
      "eval_dim_128_cosine_ndcg@10": 0.488553211966058,
      "eval_dim_128_cosine_precision@1": 0.3225491563240831,
      "eval_dim_128_cosine_precision@10": 0.07071538139729466,
      "eval_dim_128_cosine_precision@3": 0.15367452238181564,
      "eval_dim_128_cosine_precision@5": 0.107544275554316,
      "eval_dim_128_cosine_recall@1": 0.3225491563240831,
      "eval_dim_128_cosine_recall@10": 0.7071538139729466,
      "eval_dim_128_cosine_recall@3": 0.46102356714544696,
      "eval_dim_128_cosine_recall@5": 0.53772137777158,
      "eval_dim_256_cosine_accuracy@1": 0.32854553060939895,
      "eval_dim_256_cosine_accuracy@10": 0.7131501882582625,
      "eval_dim_256_cosine_accuracy@3": 0.464928182959141,
      "eval_dim_256_cosine_accuracy@5": 0.5421837958443732,
      "eval_dim_256_cosine_map@100": 0.4374891918057777,
      "eval_dim_256_cosine_mrr@10": 0.42779830799981505,
      "eval_dim_256_cosine_ndcg@10": 0.49410729071963905,
      "eval_dim_256_cosine_precision@1": 0.32854553060939895,
      "eval_dim_256_cosine_precision@10": 0.07131501882582625,
      "eval_dim_256_cosine_precision@3": 0.1549760609863803,
      "eval_dim_256_cosine_precision@5": 0.10843675916887462,
      "eval_dim_256_cosine_recall@1": 0.32854553060939895,
      "eval_dim_256_cosine_recall@10": 0.7131501882582625,
      "eval_dim_256_cosine_recall@3": 0.464928182959141,
      "eval_dim_256_cosine_recall@5": 0.5421837958443732,
      "eval_dim_512_cosine_accuracy@1": 0.33286849811741737,
      "eval_dim_512_cosine_accuracy@10": 0.7142657927764607,
      "eval_dim_512_cosine_accuracy@3": 0.46966950216148373,
      "eval_dim_512_cosine_accuracy@5": 0.5445544554455446,
      "eval_dim_512_cosine_map@100": 0.4410220731771345,
      "eval_dim_512_cosine_mrr@10": 0.4314777000838923,
      "eval_dim_512_cosine_ndcg@10": 0.4972638536904392,
      "eval_dim_512_cosine_precision@1": 0.33286849811741737,
      "eval_dim_512_cosine_precision@10": 0.07142657927764608,
      "eval_dim_512_cosine_precision@3": 0.15655650072049457,
      "eval_dim_512_cosine_precision@5": 0.10891089108910891,
      "eval_dim_512_cosine_recall@1": 0.33286849811741737,
      "eval_dim_512_cosine_recall@10": 0.7142657927764607,
      "eval_dim_512_cosine_recall@3": 0.46966950216148373,
      "eval_dim_512_cosine_recall@5": 0.5445544554455446,
      "eval_dim_64_cosine_accuracy@1": 0.306233440245433,
      "eval_dim_64_cosine_accuracy@10": 0.6906986473295217,
      "eval_dim_64_cosine_accuracy@3": 0.44205829033607585,
      "eval_dim_64_cosine_accuracy@5": 0.5208478594338307,
      "eval_dim_64_cosine_map@100": 0.416048690892794,
      "eval_dim_64_cosine_mrr@10": 0.4057003738603239,
      "eval_dim_64_cosine_ndcg@10": 0.47195343698452114,
      "eval_dim_64_cosine_precision@1": 0.306233440245433,
      "eval_dim_64_cosine_precision@10": 0.06906986473295217,
      "eval_dim_64_cosine_precision@3": 0.1473527634453586,
      "eval_dim_64_cosine_precision@5": 0.10416957188676614,
      "eval_dim_64_cosine_recall@1": 0.306233440245433,
      "eval_dim_64_cosine_recall@10": 0.6906986473295217,
      "eval_dim_64_cosine_recall@3": 0.44205829033607585,
      "eval_dim_64_cosine_recall@5": 0.5208478594338307,
      "eval_dim_768_cosine_accuracy@1": 0.33412355320039044,
      "eval_dim_768_cosine_accuracy@10": 0.7152419467298843,
      "eval_dim_768_cosine_accuracy@3": 0.47106400780923163,
      "eval_dim_768_cosine_accuracy@5": 0.5448333565750941,
      "eval_dim_768_cosine_map@100": 0.44207831994901653,
      "eval_dim_768_cosine_mrr@10": 0.4325288363846451,
      "eval_dim_768_cosine_ndcg@10": 0.49828039957821635,
      "eval_dim_768_cosine_precision@1": 0.33412355320039044,
      "eval_dim_768_cosine_precision@10": 0.07152419467298843,
      "eval_dim_768_cosine_precision@3": 0.15702133593641054,
      "eval_dim_768_cosine_precision@5": 0.10896667131501885,
      "eval_dim_768_cosine_recall@1": 0.33412355320039044,
      "eval_dim_768_cosine_recall@10": 0.7152419467298843,
      "eval_dim_768_cosine_recall@3": 0.47106400780923163,
      "eval_dim_768_cosine_recall@5": 0.5448333565750941,
      "eval_loss": 0.18898583948612213,
      "eval_runtime": 4757.4,
      "eval_samples_per_second": 1.508,
      "eval_sequential_score": 0.47195343698452114,
      "eval_steps_per_second": 0.019,
      "step": 120
    },
    {
      "epoch": 3.0,
      "grad_norm": 792357.3125,
      "learning_rate": 1.8730348307472828e-05,
      "loss": 0.1032,
      "step": 180
    },
    {
      "epoch": 3.0,
      "eval_dim_128_cosine_accuracy@1": 0.3468135545948961,
      "eval_dim_128_cosine_accuracy@10": 0.7173337052015061,
      "eval_dim_128_cosine_accuracy@3": 0.46520708408869055,
      "eval_dim_128_cosine_accuracy@5": 0.5407892901966252,
      "eval_dim_128_cosine_map@100": 0.447677926706174,
      "eval_dim_128_cosine_mrr@10": 0.43798207285517343,
      "eval_dim_128_cosine_ndcg@10": 0.5026334575249692,
      "eval_dim_128_cosine_precision@1": 0.3468135545948961,
      "eval_dim_128_cosine_precision@10": 0.07173337052015061,
      "eval_dim_128_cosine_precision@3": 0.15506902802956352,
      "eval_dim_128_cosine_precision@5": 0.10815785803932505,
      "eval_dim_128_cosine_recall@1": 0.3468135545948961,
      "eval_dim_128_cosine_recall@10": 0.7173337052015061,
      "eval_dim_128_cosine_recall@3": 0.46520708408869055,
      "eval_dim_128_cosine_recall@5": 0.5407892901966252,
      "eval_dim_256_cosine_accuracy@1": 0.35043926927904057,
      "eval_dim_256_cosine_accuracy@10": 0.7156602984242086,
      "eval_dim_256_cosine_accuracy@3": 0.4700878538558081,
      "eval_dim_256_cosine_accuracy@5": 0.544972807139869,
      "eval_dim_256_cosine_map@100": 0.4507378853512356,
      "eval_dim_256_cosine_mrr@10": 0.44098468699988747,
      "eval_dim_256_cosine_ndcg@10": 0.5046872072161374,
      "eval_dim_256_cosine_precision@1": 0.35043926927904057,
      "eval_dim_256_cosine_precision@10": 0.07156602984242087,
      "eval_dim_256_cosine_precision@3": 0.15669595128526936,
      "eval_dim_256_cosine_precision@5": 0.1089945614279738,
      "eval_dim_256_cosine_recall@1": 0.35043926927904057,
      "eval_dim_256_cosine_recall@10": 0.7156602984242086,
      "eval_dim_256_cosine_recall@3": 0.4700878538558081,
      "eval_dim_256_cosine_recall@5": 0.544972807139869,
      "eval_dim_512_cosine_accuracy@1": 0.35169432436201364,
      "eval_dim_512_cosine_accuracy@10": 0.7231906289220471,
      "eval_dim_512_cosine_accuracy@3": 0.4700878538558081,
      "eval_dim_512_cosine_accuracy@5": 0.5448333565750941,
      "eval_dim_512_cosine_map@100": 0.45212818188900633,
      "eval_dim_512_cosine_mrr@10": 0.44279272997722396,
      "eval_dim_512_cosine_ndcg@10": 0.5076751608428753,
      "eval_dim_512_cosine_precision@1": 0.35169432436201364,
      "eval_dim_512_cosine_precision@10": 0.07231906289220472,
      "eval_dim_512_cosine_precision@3": 0.15669595128526936,
      "eval_dim_512_cosine_precision@5": 0.10896667131501885,
      "eval_dim_512_cosine_recall@1": 0.35169432436201364,
      "eval_dim_512_cosine_recall@10": 0.7231906289220471,
      "eval_dim_512_cosine_recall@3": 0.4700878538558081,
      "eval_dim_512_cosine_recall@5": 0.5448333565750941,
      "eval_dim_64_cosine_accuracy@1": 0.3356575094129131,
      "eval_dim_64_cosine_accuracy@10": 0.6976711755682611,
      "eval_dim_64_cosine_accuracy@3": 0.4523776321294101,
      "eval_dim_64_cosine_accuracy@5": 0.5307488495328406,
      "eval_dim_64_cosine_map@100": 0.43559811808588345,
      "eval_dim_64_cosine_mrr@10": 0.42538852476797007,
      "eval_dim_64_cosine_ndcg@10": 0.48852868771634256,
      "eval_dim_64_cosine_precision@1": 0.3356575094129131,
      "eval_dim_64_cosine_precision@10": 0.0697671175568261,
      "eval_dim_64_cosine_precision@3": 0.1507925440431367,
      "eval_dim_64_cosine_precision@5": 0.10614976990656813,
      "eval_dim_64_cosine_recall@1": 0.3356575094129131,
      "eval_dim_64_cosine_recall@10": 0.6976711755682611,
      "eval_dim_64_cosine_recall@3": 0.4523776321294101,
      "eval_dim_64_cosine_recall@5": 0.5307488495328406,
      "eval_dim_768_cosine_accuracy@1": 0.3568539952586808,
      "eval_dim_768_cosine_accuracy@10": 0.7237484311811463,
      "eval_dim_768_cosine_accuracy@3": 0.473434667410403,
      "eval_dim_768_cosine_accuracy@5": 0.5530609398968066,
      "eval_dim_768_cosine_map@100": 0.456489701012935,
      "eval_dim_768_cosine_mrr@10": 0.44698465822437466,
      "eval_dim_768_cosine_ndcg@10": 0.5110884733352402,
      "eval_dim_768_cosine_precision@1": 0.3568539952586808,
      "eval_dim_768_cosine_precision@10": 0.07237484311811462,
      "eval_dim_768_cosine_precision@3": 0.15781155580346765,
      "eval_dim_768_cosine_precision@5": 0.11061218797936133,
      "eval_dim_768_cosine_recall@1": 0.3568539952586808,
      "eval_dim_768_cosine_recall@10": 0.7237484311811463,
      "eval_dim_768_cosine_recall@3": 0.473434667410403,
      "eval_dim_768_cosine_recall@5": 0.5530609398968066,
      "eval_loss": 0.15947215259075165,
      "eval_runtime": 4772.3881,
      "eval_samples_per_second": 1.503,
      "eval_sequential_score": 0.48852868771634256,
      "eval_steps_per_second": 0.019,
      "step": 180
    },
    {
      "epoch": 4.0,
      "grad_norm": 528732.6875,
      "learning_rate": 9.478131709729831e-06,
      "loss": 0.0562,
      "step": 240
    },
    {
      "epoch": 4.0,
      "eval_dim_128_cosine_accuracy@1": 0.35016036814949103,
      "eval_dim_128_cosine_accuracy@10": 0.71803095802538,
      "eval_dim_128_cosine_accuracy@3": 0.466880490865988,
      "eval_dim_128_cosine_accuracy@5": 0.5453911588341933,
      "eval_dim_128_cosine_map@100": 0.45032612042753656,
      "eval_dim_128_cosine_mrr@10": 0.44050070500007754,
      "eval_dim_128_cosine_ndcg@10": 0.5047593926261945,
      "eval_dim_128_cosine_precision@1": 0.35016036814949103,
      "eval_dim_128_cosine_precision@10": 0.07180309580253802,
      "eval_dim_128_cosine_precision@3": 0.15562683028866267,
      "eval_dim_128_cosine_precision@5": 0.10907823176683866,
      "eval_dim_128_cosine_recall@1": 0.35016036814949103,
      "eval_dim_128_cosine_recall@10": 0.71803095802538,
      "eval_dim_128_cosine_recall@3": 0.466880490865988,
      "eval_dim_128_cosine_recall@5": 0.5453911588341933,
      "eval_dim_256_cosine_accuracy@1": 0.34932366476084226,
      "eval_dim_256_cosine_accuracy@10": 0.7227722772277227,
      "eval_dim_256_cosine_accuracy@3": 0.46660158973643845,
      "eval_dim_256_cosine_accuracy@5": 0.5445544554455446,
      "eval_dim_256_cosine_map@100": 0.4503116262429447,
      "eval_dim_256_cosine_mrr@10": 0.44069499505282583,
      "eval_dim_256_cosine_ndcg@10": 0.5059436181895386,
      "eval_dim_256_cosine_precision@1": 0.34932366476084226,
      "eval_dim_256_cosine_precision@10": 0.07227722772277227,
      "eval_dim_256_cosine_precision@3": 0.15553386324547946,
      "eval_dim_256_cosine_precision@5": 0.10891089108910891,
      "eval_dim_256_cosine_recall@1": 0.34932366476084226,
      "eval_dim_256_cosine_recall@10": 0.7227722772277227,
      "eval_dim_256_cosine_recall@3": 0.46660158973643845,
      "eval_dim_256_cosine_recall@5": 0.5445544554455446,
      "eval_dim_512_cosine_accuracy@1": 0.3550411379166086,
      "eval_dim_512_cosine_accuracy@10": 0.7234695300515968,
      "eval_dim_512_cosine_accuracy@3": 0.4693906010319342,
      "eval_dim_512_cosine_accuracy@5": 0.5484590712592387,
      "eval_dim_512_cosine_map@100": 0.45428241375668854,
      "eval_dim_512_cosine_mrr@10": 0.4447460339595335,
      "eval_dim_512_cosine_ndcg@10": 0.5091968606671677,
      "eval_dim_512_cosine_precision@1": 0.3550411379166086,
      "eval_dim_512_cosine_precision@10": 0.07234695300515968,
      "eval_dim_512_cosine_precision@3": 0.1564635336773114,
      "eval_dim_512_cosine_precision@5": 0.10969181425184772,
      "eval_dim_512_cosine_recall@1": 0.3550411379166086,
      "eval_dim_512_cosine_recall@10": 0.7234695300515968,
      "eval_dim_512_cosine_recall@3": 0.4693906010319342,
      "eval_dim_512_cosine_recall@5": 0.5484590712592387,
      "eval_dim_64_cosine_accuracy@1": 0.33760981731976014,
      "eval_dim_64_cosine_accuracy@10": 0.7082694184911449,
      "eval_dim_64_cosine_accuracy@3": 0.45349323664760843,
      "eval_dim_64_cosine_accuracy@5": 0.5308883000976153,
      "eval_dim_64_cosine_map@100": 0.4383018870756752,
      "eval_dim_64_cosine_mrr@10": 0.4285745828103948,
      "eval_dim_64_cosine_ndcg@10": 0.49330694449266793,
      "eval_dim_64_cosine_precision@1": 0.33760981731976014,
      "eval_dim_64_cosine_precision@10": 0.07082694184911449,
      "eval_dim_64_cosine_precision@3": 0.1511644122158695,
      "eval_dim_64_cosine_precision@5": 0.10617766001952307,
      "eval_dim_64_cosine_recall@1": 0.33760981731976014,
      "eval_dim_64_cosine_recall@10": 0.7082694184911449,
      "eval_dim_64_cosine_recall@3": 0.45349323664760843,
      "eval_dim_64_cosine_recall@5": 0.5308883000976153,
      "eval_dim_768_cosine_accuracy@1": 0.36061916050760007,
      "eval_dim_768_cosine_accuracy@10": 0.7238878817459211,
      "eval_dim_768_cosine_accuracy@3": 0.478036536047971,
      "eval_dim_768_cosine_accuracy@5": 0.5529214893320318,
      "eval_dim_768_cosine_map@100": 0.45954079808905995,
      "eval_dim_768_cosine_mrr@10": 0.44983946583793183,
      "eval_dim_768_cosine_ndcg@10": 0.513288052386553,
      "eval_dim_768_cosine_precision@1": 0.36061916050760007,
      "eval_dim_768_cosine_precision@10": 0.07238878817459211,
      "eval_dim_768_cosine_precision@3": 0.15934551201599031,
      "eval_dim_768_cosine_precision@5": 0.11058429786640636,
      "eval_dim_768_cosine_recall@1": 0.36061916050760007,
      "eval_dim_768_cosine_recall@10": 0.7238878817459211,
      "eval_dim_768_cosine_recall@3": 0.478036536047971,
      "eval_dim_768_cosine_recall@5": 0.5529214893320318,
      "eval_loss": 0.14603222906589508,
      "eval_runtime": 4765.1388,
      "eval_samples_per_second": 1.505,
      "eval_sequential_score": 0.49330694449266793,
      "eval_steps_per_second": 0.019,
      "step": 240
    }
  ],
  "logging_steps": 500,
  "max_steps": 354,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 6,
  "save_steps": 500,
  "stateful_callbacks": {
    "EarlyStoppingCallback": {
      "args": {
        "early_stopping_patience": 1,
        "early_stopping_threshold": 0.001
      },
      "attributes": {
        "early_stopping_patience_counter": 0
      }
    },
    "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": 80,
  "trial_name": null,
  "trial_params": null
}