{
  "dataset_name": "ToT-Math-V1",
  "group_index": 2,
  "prompt_group_file": "/hkfs/work/workspace/scratch/tum_fmp0582-dndworkspace/自己训练lora/train_lora/prompt_groups/ToT-Math-V1/group_02.json",
  "output_dir": "/hkfs/work/workspace/scratch/tum_fmp0582-dndworkspace/自己训练lora/train_lora/outputs/ToT-Math-V1/group_02",
  "checkpoint_root": "/hkfs/work/workspace/scratch/tum_fmp0582-dndworkspace/自己训练lora/train_lora/outputs/ToT-Math-V1/group_02/checkpoints",
  "generated_at": "2025-11-06T12:08:05Z",
  "train_loss": 0.04049773897975683,
  "metrics": {
    "train_runtime": 30699.1941,
    "train_samples_per_second": 1.876,
    "train_steps_per_second": 0.059,
    "total_flos": 1.5308141101056e+18,
    "train_loss": 0.04049773897975683,
    "epoch": 450.0
  },
  "trainer_state": [
    {
      "loss": 1.3768,
      "grad_norm": 3.3850419521331787,
      "learning_rate": 0.0,
      "epoch": 0.32,
      "step": 1
    },
    {
      "loss": 1.4153,
      "grad_norm": 2.2271716594696045,
      "learning_rate": 7.4074074074074075e-06,
      "epoch": 1.32,
      "step": 5
    },
    {
      "loss": 1.3367,
      "grad_norm": 1.1049851179122925,
      "learning_rate": 1.6666666666666667e-05,
      "epoch": 2.64,
      "step": 10
    },
    {
      "loss": 1.1976,
      "grad_norm": 0.6343593001365662,
      "learning_rate": 2.5925925925925925e-05,
      "epoch": 3.96,
      "step": 15
    },
    {
      "loss": 0.9942,
      "grad_norm": 0.5920177102088928,
      "learning_rate": 3.518518518518519e-05,
      "epoch": 5.0,
      "step": 20
    },
    {
      "loss": 1.0356,
      "grad_norm": 0.3647186756134033,
      "learning_rate": 4.4444444444444447e-05,
      "epoch": 6.32,
      "step": 25
    },
    {
      "loss": 0.9349,
      "grad_norm": 0.32298368215560913,
      "learning_rate": 5.370370370370371e-05,
      "epoch": 7.64,
      "step": 30
    },
    {
      "loss": 0.81,
      "grad_norm": 0.3340265154838562,
      "learning_rate": 6.296296296296296e-05,
      "epoch": 8.96,
      "step": 35
    },
    {
      "loss": 0.7693,
      "grad_norm": 0.9156715869903564,
      "learning_rate": 7.222222222222222e-05,
      "epoch": 10.0,
      "step": 40
    },
    {
      "loss": 0.6753,
      "grad_norm": 0.3196317255496979,
      "learning_rate": 8.148148148148148e-05,
      "epoch": 11.32,
      "step": 45
    },
    {
      "loss": 0.6121,
      "grad_norm": 0.3152695596218109,
      "learning_rate": 9.074074074074075e-05,
      "epoch": 12.64,
      "step": 50
    },
    {
      "loss": 0.5313,
      "grad_norm": 0.4185727834701538,
      "learning_rate": 0.0001,
      "epoch": 13.96,
      "step": 55
    },
    {
      "loss": 0.4546,
      "grad_norm": 1.2220144271850586,
      "learning_rate": 9.971363115693013e-05,
      "epoch": 15.0,
      "step": 60
    },
    {
      "loss": 0.3208,
      "grad_norm": 0.42747265100479126,
      "learning_rate": 9.942726231386026e-05,
      "epoch": 16.32,
      "step": 65
    },
    {
      "loss": 0.245,
      "grad_norm": 0.44386783242225647,
      "learning_rate": 9.914089347079038e-05,
      "epoch": 17.64,
      "step": 70
    },
    {
      "loss": 0.1586,
      "grad_norm": 0.5088019371032715,
      "learning_rate": 9.885452462772051e-05,
      "epoch": 18.96,
      "step": 75
    },
    {
      "loss": 0.1022,
      "grad_norm": 1.5268586874008179,
      "learning_rate": 9.856815578465064e-05,
      "epoch": 20.0,
      "step": 80
    },
    {
      "loss": 0.0668,
      "grad_norm": 0.38557520508766174,
      "learning_rate": 9.828178694158075e-05,
      "epoch": 21.32,
      "step": 85
    },
    {
      "loss": 0.0462,
      "grad_norm": 0.43242698907852173,
      "learning_rate": 9.799541809851088e-05,
      "epoch": 22.64,
      "step": 90
    },
    {
      "loss": 0.0388,
      "grad_norm": 0.4117985665798187,
      "learning_rate": 9.7709049255441e-05,
      "epoch": 23.96,
      "step": 95
    },
    {
      "loss": 0.0321,
      "grad_norm": 0.8697754144668579,
      "learning_rate": 9.742268041237114e-05,
      "epoch": 25.0,
      "step": 100
    },
    {
      "loss": 0.0254,
      "grad_norm": 0.2642986476421356,
      "learning_rate": 9.713631156930127e-05,
      "epoch": 26.32,
      "step": 105
    },
    {
      "loss": 0.0223,
      "grad_norm": 0.3619579076766968,
      "learning_rate": 9.68499427262314e-05,
      "epoch": 27.64,
      "step": 110
    },
    {
      "loss": 0.0202,
      "grad_norm": 0.2517232298851013,
      "learning_rate": 9.656357388316152e-05,
      "epoch": 28.96,
      "step": 115
    },
    {
      "loss": 0.018,
      "grad_norm": 0.4181807339191437,
      "learning_rate": 9.627720504009165e-05,
      "epoch": 30.0,
      "step": 120
    },
    {
      "loss": 0.0161,
      "grad_norm": 0.12266844511032104,
      "learning_rate": 9.599083619702178e-05,
      "epoch": 31.32,
      "step": 125
    },
    {
      "loss": 0.0146,
      "grad_norm": 0.1561511754989624,
      "learning_rate": 9.57044673539519e-05,
      "epoch": 32.64,
      "step": 130
    },
    {
      "loss": 0.0125,
      "grad_norm": 0.1291125863790512,
      "learning_rate": 9.541809851088203e-05,
      "epoch": 33.96,
      "step": 135
    },
    {
      "loss": 0.0125,
      "grad_norm": 0.9811831712722778,
      "learning_rate": 9.513172966781214e-05,
      "epoch": 35.0,
      "step": 140
    },
    {
      "loss": 0.0129,
      "grad_norm": 0.12179408967494965,
      "learning_rate": 9.484536082474227e-05,
      "epoch": 36.32,
      "step": 145
    },
    {
      "loss": 0.0131,
      "grad_norm": 0.18377508223056793,
      "learning_rate": 9.45589919816724e-05,
      "epoch": 37.64,
      "step": 150
    },
    {
      "loss": 0.0112,
      "grad_norm": 0.10695623606443405,
      "learning_rate": 9.427262313860252e-05,
      "epoch": 38.96,
      "step": 155
    },
    {
      "loss": 0.0116,
      "grad_norm": 0.09302873909473419,
      "learning_rate": 9.398625429553265e-05,
      "epoch": 40.0,
      "step": 160
    },
    {
      "loss": 0.011,
      "grad_norm": 0.041765324771404266,
      "learning_rate": 9.369988545246277e-05,
      "epoch": 41.32,
      "step": 165
    },
    {
      "loss": 0.01,
      "grad_norm": 0.06779934465885162,
      "learning_rate": 9.34135166093929e-05,
      "epoch": 42.64,
      "step": 170
    },
    {
      "loss": 0.0101,
      "grad_norm": 0.061006441712379456,
      "learning_rate": 9.312714776632303e-05,
      "epoch": 43.96,
      "step": 175
    },
    {
      "loss": 0.0092,
      "grad_norm": 0.06570079177618027,
      "learning_rate": 9.284077892325315e-05,
      "epoch": 45.0,
      "step": 180
    },
    {
      "loss": 0.0096,
      "grad_norm": 0.03699507564306259,
      "learning_rate": 9.255441008018328e-05,
      "epoch": 46.32,
      "step": 185
    },
    {
      "loss": 0.0091,
      "grad_norm": 0.03686221316456795,
      "learning_rate": 9.22680412371134e-05,
      "epoch": 47.64,
      "step": 190
    },
    {
      "loss": 0.0102,
      "grad_norm": 0.2509954571723938,
      "learning_rate": 9.198167239404353e-05,
      "epoch": 48.96,
      "step": 195
    },
    {
      "loss": 0.0096,
      "grad_norm": 0.12955909967422485,
      "learning_rate": 9.169530355097366e-05,
      "epoch": 50.0,
      "step": 200
    },
    {
      "loss": 0.0088,
      "grad_norm": 0.04445841908454895,
      "learning_rate": 9.140893470790379e-05,
      "epoch": 51.32,
      "step": 205
    },
    {
      "loss": 0.0092,
      "grad_norm": 0.03362618014216423,
      "learning_rate": 9.112256586483391e-05,
      "epoch": 52.64,
      "step": 210
    },
    {
      "loss": 0.0088,
      "grad_norm": 0.07259737700223923,
      "learning_rate": 9.083619702176404e-05,
      "epoch": 53.96,
      "step": 215
    },
    {
      "loss": 0.009,
      "grad_norm": 0.1037302017211914,
      "learning_rate": 9.054982817869416e-05,
      "epoch": 55.0,
      "step": 220
    },
    {
      "loss": 0.0092,
      "grad_norm": 0.03083176352083683,
      "learning_rate": 9.026345933562429e-05,
      "epoch": 56.32,
      "step": 225
    },
    {
      "loss": 0.009,
      "grad_norm": 0.031119555234909058,
      "learning_rate": 8.997709049255442e-05,
      "epoch": 57.64,
      "step": 230
    },
    {
      "loss": 0.0101,
      "grad_norm": 0.028364112600684166,
      "learning_rate": 8.969072164948454e-05,
      "epoch": 58.96,
      "step": 235
    },
    {
      "loss": 0.0094,
      "grad_norm": 0.08577492088079453,
      "learning_rate": 8.940435280641467e-05,
      "epoch": 60.0,
      "step": 240
    },
    {
      "loss": 0.0086,
      "grad_norm": 0.028203085064888,
      "learning_rate": 8.91179839633448e-05,
      "epoch": 61.32,
      "step": 245
    },
    {
      "loss": 0.0089,
      "grad_norm": 0.032198961824178696,
      "learning_rate": 8.883161512027491e-05,
      "epoch": 62.64,
      "step": 250
    },
    {
      "loss": 0.0088,
      "grad_norm": 0.02721354365348816,
      "learning_rate": 8.854524627720504e-05,
      "epoch": 63.96,
      "step": 255
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.07893114537000656,
      "learning_rate": 8.825887743413516e-05,
      "epoch": 65.0,
      "step": 260
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.02526146173477173,
      "learning_rate": 8.797250859106529e-05,
      "epoch": 66.32,
      "step": 265
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.025326905772089958,
      "learning_rate": 8.768613974799542e-05,
      "epoch": 67.64,
      "step": 270
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.02333327755331993,
      "learning_rate": 8.739977090492554e-05,
      "epoch": 68.96,
      "step": 275
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.07976648956537247,
      "learning_rate": 8.711340206185567e-05,
      "epoch": 70.0,
      "step": 280
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.023474574089050293,
      "learning_rate": 8.682703321878581e-05,
      "epoch": 71.32,
      "step": 285
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.025870641693472862,
      "learning_rate": 8.654066437571594e-05,
      "epoch": 72.64,
      "step": 290
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.027549639344215393,
      "learning_rate": 8.625429553264606e-05,
      "epoch": 73.96,
      "step": 295
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.050593771040439606,
      "learning_rate": 8.596792668957619e-05,
      "epoch": 75.0,
      "step": 300
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.022557204589247704,
      "learning_rate": 8.56815578465063e-05,
      "epoch": 76.32,
      "step": 305
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.024449430406093597,
      "learning_rate": 8.539518900343643e-05,
      "epoch": 77.64,
      "step": 310
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.02718745730817318,
      "learning_rate": 8.510882016036655e-05,
      "epoch": 78.96,
      "step": 315
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.07632927596569061,
      "learning_rate": 8.482245131729668e-05,
      "epoch": 80.0,
      "step": 320
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.020235633477568626,
      "learning_rate": 8.453608247422681e-05,
      "epoch": 81.32,
      "step": 325
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.02495831996202469,
      "learning_rate": 8.424971363115693e-05,
      "epoch": 82.64,
      "step": 330
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.028957359492778778,
      "learning_rate": 8.396334478808706e-05,
      "epoch": 83.96,
      "step": 335
    },
    {
      "loss": 0.0085,
      "grad_norm": 0.1061309427022934,
      "learning_rate": 8.367697594501719e-05,
      "epoch": 85.0,
      "step": 340
    },
    {
      "loss": 0.008,
      "grad_norm": 0.02090873382985592,
      "learning_rate": 8.339060710194731e-05,
      "epoch": 86.32,
      "step": 345
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.024526020511984825,
      "learning_rate": 8.310423825887744e-05,
      "epoch": 87.64,
      "step": 350
    },
    {
      "loss": 0.0085,
      "grad_norm": 0.029449045658111572,
      "learning_rate": 8.281786941580757e-05,
      "epoch": 88.96,
      "step": 355
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.052238401025533676,
      "learning_rate": 8.253150057273768e-05,
      "epoch": 90.0,
      "step": 360
    },
    {
      "loss": 0.0085,
      "grad_norm": 0.021808411926031113,
      "learning_rate": 8.224513172966782e-05,
      "epoch": 91.32,
      "step": 365
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.021965432912111282,
      "learning_rate": 8.195876288659795e-05,
      "epoch": 92.64,
      "step": 370
    },
    {
      "loss": 0.008,
      "grad_norm": 0.02108742482960224,
      "learning_rate": 8.167239404352807e-05,
      "epoch": 93.96,
      "step": 375
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.0818358063697815,
      "learning_rate": 8.13860252004582e-05,
      "epoch": 95.0,
      "step": 380
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.023490387946367264,
      "learning_rate": 8.109965635738833e-05,
      "epoch": 96.32,
      "step": 385
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.021999714896082878,
      "learning_rate": 8.081328751431845e-05,
      "epoch": 97.64,
      "step": 390
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.02380586601793766,
      "learning_rate": 8.052691867124858e-05,
      "epoch": 98.96,
      "step": 395
    },
    {
      "loss": 0.0091,
      "grad_norm": 0.08215590566396713,
      "learning_rate": 8.02405498281787e-05,
      "epoch": 100.0,
      "step": 400
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.022699300199747086,
      "learning_rate": 7.995418098510883e-05,
      "epoch": 101.32,
      "step": 405
    },
    {
      "loss": 0.0086,
      "grad_norm": 0.024920256808400154,
      "learning_rate": 7.966781214203894e-05,
      "epoch": 102.64,
      "step": 410
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.0218623299151659,
      "learning_rate": 7.938144329896907e-05,
      "epoch": 103.96,
      "step": 415
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.08335137367248535,
      "learning_rate": 7.90950744558992e-05,
      "epoch": 105.0,
      "step": 420
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.019201533868908882,
      "learning_rate": 7.880870561282932e-05,
      "epoch": 106.32,
      "step": 425
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.017693670466542244,
      "learning_rate": 7.852233676975945e-05,
      "epoch": 107.64,
      "step": 430
    },
    {
      "loss": 0.008,
      "grad_norm": 0.023311873897910118,
      "learning_rate": 7.823596792668958e-05,
      "epoch": 108.96,
      "step": 435
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.06780176609754562,
      "learning_rate": 7.79495990836197e-05,
      "epoch": 110.0,
      "step": 440
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.024123620241880417,
      "learning_rate": 7.766323024054983e-05,
      "epoch": 111.32,
      "step": 445
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.018307629972696304,
      "learning_rate": 7.737686139747996e-05,
      "epoch": 112.64,
      "step": 450
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.018539734184741974,
      "learning_rate": 7.709049255441008e-05,
      "epoch": 113.96,
      "step": 455
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.05340789631009102,
      "learning_rate": 7.680412371134021e-05,
      "epoch": 115.0,
      "step": 460
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.020492475479841232,
      "learning_rate": 7.651775486827034e-05,
      "epoch": 116.32,
      "step": 465
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.019051015377044678,
      "learning_rate": 7.623138602520046e-05,
      "epoch": 117.64,
      "step": 470
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.023864563554525375,
      "learning_rate": 7.594501718213059e-05,
      "epoch": 118.96,
      "step": 475
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.0733332484960556,
      "learning_rate": 7.565864833906071e-05,
      "epoch": 120.0,
      "step": 480
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.019676174968481064,
      "learning_rate": 7.537227949599084e-05,
      "epoch": 121.32,
      "step": 485
    },
    {
      "loss": 0.008,
      "grad_norm": 0.019537003710865974,
      "learning_rate": 7.508591065292097e-05,
      "epoch": 122.64,
      "step": 490
    },
    {
      "loss": 0.0091,
      "grad_norm": 0.021838398650288582,
      "learning_rate": 7.47995418098511e-05,
      "epoch": 123.96,
      "step": 495
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.058284178376197815,
      "learning_rate": 7.451317296678122e-05,
      "epoch": 125.0,
      "step": 500
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.020997829735279083,
      "learning_rate": 7.422680412371135e-05,
      "epoch": 126.32,
      "step": 505
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.023058714345097542,
      "learning_rate": 7.394043528064147e-05,
      "epoch": 127.64,
      "step": 510
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.021496370434761047,
      "learning_rate": 7.36540664375716e-05,
      "epoch": 128.96,
      "step": 515
    },
    {
      "loss": 0.0085,
      "grad_norm": 0.05587420612573624,
      "learning_rate": 7.336769759450171e-05,
      "epoch": 130.0,
      "step": 520
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.019805986434221268,
      "learning_rate": 7.308132875143184e-05,
      "epoch": 131.32,
      "step": 525
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.01775645837187767,
      "learning_rate": 7.279495990836197e-05,
      "epoch": 132.64,
      "step": 530
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.02013344317674637,
      "learning_rate": 7.250859106529209e-05,
      "epoch": 133.96,
      "step": 535
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.05250405892729759,
      "learning_rate": 7.222222222222222e-05,
      "epoch": 135.0,
      "step": 540
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.019977595657110214,
      "learning_rate": 7.193585337915235e-05,
      "epoch": 136.32,
      "step": 545
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.018704729154706,
      "learning_rate": 7.164948453608247e-05,
      "epoch": 137.64,
      "step": 550
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.01925542578101158,
      "learning_rate": 7.136311569301261e-05,
      "epoch": 138.96,
      "step": 555
    },
    {
      "loss": 0.0086,
      "grad_norm": 0.06180017068982124,
      "learning_rate": 7.107674684994274e-05,
      "epoch": 140.0,
      "step": 560
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.019194940105080605,
      "learning_rate": 7.079037800687286e-05,
      "epoch": 141.32,
      "step": 565
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.01840827614068985,
      "learning_rate": 7.050400916380299e-05,
      "epoch": 142.64,
      "step": 570
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.017798591405153275,
      "learning_rate": 7.02176403207331e-05,
      "epoch": 143.96,
      "step": 575
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.07515519857406616,
      "learning_rate": 6.993127147766323e-05,
      "epoch": 145.0,
      "step": 580
    },
    {
      "loss": 0.0085,
      "grad_norm": 0.018694644793868065,
      "learning_rate": 6.964490263459336e-05,
      "epoch": 146.32,
      "step": 585
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.02003779634833336,
      "learning_rate": 6.935853379152348e-05,
      "epoch": 147.64,
      "step": 590
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.019536534324288368,
      "learning_rate": 6.907216494845361e-05,
      "epoch": 148.96,
      "step": 595
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.05964312329888344,
      "learning_rate": 6.878579610538374e-05,
      "epoch": 150.0,
      "step": 600
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.01586931198835373,
      "learning_rate": 6.849942726231386e-05,
      "epoch": 151.32,
      "step": 605
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.019150903448462486,
      "learning_rate": 6.821305841924399e-05,
      "epoch": 152.64,
      "step": 610
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.01623143069446087,
      "learning_rate": 6.792668957617412e-05,
      "epoch": 153.96,
      "step": 615
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.05439450591802597,
      "learning_rate": 6.764032073310424e-05,
      "epoch": 155.0,
      "step": 620
    },
    {
      "loss": 0.008,
      "grad_norm": 0.015493949875235558,
      "learning_rate": 6.735395189003437e-05,
      "epoch": 156.32,
      "step": 625
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.015258340165019035,
      "learning_rate": 6.706758304696448e-05,
      "epoch": 157.64,
      "step": 630
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.0177009180188179,
      "learning_rate": 6.678121420389462e-05,
      "epoch": 158.96,
      "step": 635
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.04660150036215782,
      "learning_rate": 6.649484536082475e-05,
      "epoch": 160.0,
      "step": 640
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.015658147633075714,
      "learning_rate": 6.620847651775487e-05,
      "epoch": 161.32,
      "step": 645
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.016052214428782463,
      "learning_rate": 6.5922107674685e-05,
      "epoch": 162.64,
      "step": 650
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.018937313929200172,
      "learning_rate": 6.563573883161513e-05,
      "epoch": 163.96,
      "step": 655
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.04806089028716087,
      "learning_rate": 6.534936998854525e-05,
      "epoch": 165.0,
      "step": 660
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.015178844332695007,
      "learning_rate": 6.506300114547538e-05,
      "epoch": 166.32,
      "step": 665
    },
    {
      "loss": 0.0088,
      "grad_norm": 0.015573147684335709,
      "learning_rate": 6.477663230240551e-05,
      "epoch": 167.64,
      "step": 670
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.019762985408306122,
      "learning_rate": 6.449026345933563e-05,
      "epoch": 168.96,
      "step": 675
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.04972008615732193,
      "learning_rate": 6.420389461626576e-05,
      "epoch": 170.0,
      "step": 680
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.015284025110304356,
      "learning_rate": 6.391752577319587e-05,
      "epoch": 171.32,
      "step": 685
    },
    {
      "loss": 0.008,
      "grad_norm": 0.017075438052415848,
      "learning_rate": 6.3631156930126e-05,
      "epoch": 172.64,
      "step": 690
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.016540363430976868,
      "learning_rate": 6.334478808705613e-05,
      "epoch": 173.96,
      "step": 695
    },
    {
      "loss": 0.008,
      "grad_norm": 0.04700656235218048,
      "learning_rate": 6.305841924398625e-05,
      "epoch": 175.0,
      "step": 700
    },
    {
      "loss": 0.008,
      "grad_norm": 0.016470735892653465,
      "learning_rate": 6.277205040091638e-05,
      "epoch": 176.32,
      "step": 705
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.018279127776622772,
      "learning_rate": 6.24856815578465e-05,
      "epoch": 177.64,
      "step": 710
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.017167337238788605,
      "learning_rate": 6.219931271477663e-05,
      "epoch": 178.96,
      "step": 715
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.06051924452185631,
      "learning_rate": 6.191294387170676e-05,
      "epoch": 180.0,
      "step": 720
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.01896810717880726,
      "learning_rate": 6.162657502863689e-05,
      "epoch": 181.32,
      "step": 725
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.015502582304179668,
      "learning_rate": 6.134020618556701e-05,
      "epoch": 182.64,
      "step": 730
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.01537818182259798,
      "learning_rate": 6.105383734249714e-05,
      "epoch": 183.96,
      "step": 735
    },
    {
      "loss": 0.0085,
      "grad_norm": 0.05605476722121239,
      "learning_rate": 6.076746849942726e-05,
      "epoch": 185.0,
      "step": 740
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.0168443750590086,
      "learning_rate": 6.0481099656357384e-05,
      "epoch": 186.32,
      "step": 745
    },
    {
      "loss": 0.008,
      "grad_norm": 0.01575298421084881,
      "learning_rate": 6.019473081328752e-05,
      "epoch": 187.64,
      "step": 750
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.017613498494029045,
      "learning_rate": 5.9908361970217644e-05,
      "epoch": 188.96,
      "step": 755
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.05125207453966141,
      "learning_rate": 5.962199312714777e-05,
      "epoch": 190.0,
      "step": 760
    },
    {
      "loss": 0.008,
      "grad_norm": 0.015345912426710129,
      "learning_rate": 5.93356242840779e-05,
      "epoch": 191.32,
      "step": 765
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.017093928530812263,
      "learning_rate": 5.904925544100802e-05,
      "epoch": 192.64,
      "step": 770
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.016741206869482994,
      "learning_rate": 5.876288659793815e-05,
      "epoch": 193.96,
      "step": 775
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.05757911503314972,
      "learning_rate": 5.8476517754868276e-05,
      "epoch": 195.0,
      "step": 780
    },
    {
      "loss": 0.008,
      "grad_norm": 0.01850571483373642,
      "learning_rate": 5.81901489117984e-05,
      "epoch": 196.32,
      "step": 785
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.016377229243516922,
      "learning_rate": 5.790378006872853e-05,
      "epoch": 197.64,
      "step": 790
    },
    {
      "loss": 0.008,
      "grad_norm": 0.01596013829112053,
      "learning_rate": 5.761741122565865e-05,
      "epoch": 198.96,
      "step": 795
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.04500488191843033,
      "learning_rate": 5.7331042382588775e-05,
      "epoch": 200.0,
      "step": 800
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.015039285644888878,
      "learning_rate": 5.70446735395189e-05,
      "epoch": 201.32,
      "step": 805
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.01673053577542305,
      "learning_rate": 5.675830469644903e-05,
      "epoch": 202.64,
      "step": 810
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.016573773697018623,
      "learning_rate": 5.6471935853379155e-05,
      "epoch": 203.96,
      "step": 815
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.04260169714689255,
      "learning_rate": 5.618556701030928e-05,
      "epoch": 205.0,
      "step": 820
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.01616034284234047,
      "learning_rate": 5.589919816723941e-05,
      "epoch": 206.32,
      "step": 825
    },
    {
      "loss": 0.008,
      "grad_norm": 0.017032507807016373,
      "learning_rate": 5.5612829324169534e-05,
      "epoch": 207.64,
      "step": 830
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.017485234886407852,
      "learning_rate": 5.532646048109966e-05,
      "epoch": 208.96,
      "step": 835
    },
    {
      "loss": 0.008,
      "grad_norm": 0.07304313033819199,
      "learning_rate": 5.504009163802979e-05,
      "epoch": 210.0,
      "step": 840
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.01685146614909172,
      "learning_rate": 5.4753722794959914e-05,
      "epoch": 211.32,
      "step": 845
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.01598404161632061,
      "learning_rate": 5.4467353951890033e-05,
      "epoch": 212.64,
      "step": 850
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.015317142941057682,
      "learning_rate": 5.418098510882016e-05,
      "epoch": 213.96,
      "step": 855
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.044980406761169434,
      "learning_rate": 5.3894616265750286e-05,
      "epoch": 215.0,
      "step": 860
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014860185794532299,
      "learning_rate": 5.360824742268041e-05,
      "epoch": 216.32,
      "step": 865
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.015746012330055237,
      "learning_rate": 5.332187857961054e-05,
      "epoch": 217.64,
      "step": 870
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.01420033909380436,
      "learning_rate": 5.3035509736540666e-05,
      "epoch": 218.96,
      "step": 875
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.04616374149918556,
      "learning_rate": 5.274914089347079e-05,
      "epoch": 220.0,
      "step": 880
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.014527595601975918,
      "learning_rate": 5.246277205040092e-05,
      "epoch": 221.32,
      "step": 885
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.013293003663420677,
      "learning_rate": 5.2176403207331045e-05,
      "epoch": 222.64,
      "step": 890
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.015138545073568821,
      "learning_rate": 5.189003436426118e-05,
      "epoch": 223.96,
      "step": 895
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.06050126999616623,
      "learning_rate": 5.1603665521191305e-05,
      "epoch": 225.0,
      "step": 900
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.01608823426067829,
      "learning_rate": 5.131729667812142e-05,
      "epoch": 226.32,
      "step": 905
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.014482674188911915,
      "learning_rate": 5.1030927835051544e-05,
      "epoch": 227.64,
      "step": 910
    },
    {
      "loss": 0.0085,
      "grad_norm": 0.017899947240948677,
      "learning_rate": 5.074455899198167e-05,
      "epoch": 228.96,
      "step": 915
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.04981916397809982,
      "learning_rate": 5.04581901489118e-05,
      "epoch": 230.0,
      "step": 920
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.014127983711659908,
      "learning_rate": 5.0171821305841924e-05,
      "epoch": 231.32,
      "step": 925
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.016168996691703796,
      "learning_rate": 4.988545246277205e-05,
      "epoch": 232.64,
      "step": 930
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.01672256924211979,
      "learning_rate": 4.9599083619702184e-05,
      "epoch": 233.96,
      "step": 935
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.05222675949335098,
      "learning_rate": 4.931271477663231e-05,
      "epoch": 235.0,
      "step": 940
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.015397349372506142,
      "learning_rate": 4.902634593356243e-05,
      "epoch": 236.32,
      "step": 945
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.01607559435069561,
      "learning_rate": 4.8739977090492556e-05,
      "epoch": 237.64,
      "step": 950
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.01579202525317669,
      "learning_rate": 4.845360824742268e-05,
      "epoch": 238.96,
      "step": 955
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.058737389743328094,
      "learning_rate": 4.816723940435281e-05,
      "epoch": 240.0,
      "step": 960
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.016746170818805695,
      "learning_rate": 4.7880870561282936e-05,
      "epoch": 241.32,
      "step": 965
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.014333747327327728,
      "learning_rate": 4.7594501718213055e-05,
      "epoch": 242.64,
      "step": 970
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.019814383238554,
      "learning_rate": 4.730813287514318e-05,
      "epoch": 243.96,
      "step": 975
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.046136464923620224,
      "learning_rate": 4.7021764032073315e-05,
      "epoch": 245.0,
      "step": 980
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015283667482435703,
      "learning_rate": 4.673539518900344e-05,
      "epoch": 246.32,
      "step": 985
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.018138457089662552,
      "learning_rate": 4.644902634593357e-05,
      "epoch": 247.64,
      "step": 990
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.01637330837547779,
      "learning_rate": 4.6162657502863694e-05,
      "epoch": 248.96,
      "step": 995
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.05327974632382393,
      "learning_rate": 4.5876288659793814e-05,
      "epoch": 250.0,
      "step": 1000
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015156740322709084,
      "learning_rate": 4.558991981672394e-05,
      "epoch": 251.32,
      "step": 1005
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.013785932213068008,
      "learning_rate": 4.530355097365407e-05,
      "epoch": 252.64,
      "step": 1010
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.015375819057226181,
      "learning_rate": 4.5017182130584194e-05,
      "epoch": 253.96,
      "step": 1015
    },
    {
      "loss": 0.0084,
      "grad_norm": 0.05033525824546814,
      "learning_rate": 4.473081328751432e-05,
      "epoch": 255.0,
      "step": 1020
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.015314307995140553,
      "learning_rate": 4.4444444444444447e-05,
      "epoch": 256.32,
      "step": 1025
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.014964088797569275,
      "learning_rate": 4.415807560137457e-05,
      "epoch": 257.64,
      "step": 1030
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.014881624840199947,
      "learning_rate": 4.38717067583047e-05,
      "epoch": 258.96,
      "step": 1035
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.04606485366821289,
      "learning_rate": 4.3585337915234826e-05,
      "epoch": 260.0,
      "step": 1040
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.013802210800349712,
      "learning_rate": 4.329896907216495e-05,
      "epoch": 261.32,
      "step": 1045
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.013128722086548805,
      "learning_rate": 4.301260022909508e-05,
      "epoch": 262.64,
      "step": 1050
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013798913918435574,
      "learning_rate": 4.27262313860252e-05,
      "epoch": 263.96,
      "step": 1055
    },
    {
      "loss": 0.0088,
      "grad_norm": 0.04967493563890457,
      "learning_rate": 4.2439862542955325e-05,
      "epoch": 265.0,
      "step": 1060
    },
    {
      "loss": 0.008,
      "grad_norm": 0.01490580290555954,
      "learning_rate": 4.215349369988545e-05,
      "epoch": 266.32,
      "step": 1065
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.01472641620784998,
      "learning_rate": 4.1867124856815585e-05,
      "epoch": 267.64,
      "step": 1070
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.017258981242775917,
      "learning_rate": 4.158075601374571e-05,
      "epoch": 268.96,
      "step": 1075
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.037954483181238174,
      "learning_rate": 4.129438717067583e-05,
      "epoch": 270.0,
      "step": 1080
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.012794313952326775,
      "learning_rate": 4.100801832760596e-05,
      "epoch": 271.32,
      "step": 1085
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.013722005300223827,
      "learning_rate": 4.0721649484536084e-05,
      "epoch": 272.64,
      "step": 1090
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.0135383615270257,
      "learning_rate": 4.043528064146621e-05,
      "epoch": 273.96,
      "step": 1095
    },
    {
      "loss": 0.0088,
      "grad_norm": 0.05128846317529678,
      "learning_rate": 4.014891179839634e-05,
      "epoch": 275.0,
      "step": 1100
    },
    {
      "loss": 0.008,
      "grad_norm": 0.01461031660437584,
      "learning_rate": 3.9862542955326463e-05,
      "epoch": 276.32,
      "step": 1105
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.01566358096897602,
      "learning_rate": 3.957617411225659e-05,
      "epoch": 277.64,
      "step": 1110
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.014194756746292114,
      "learning_rate": 3.9289805269186716e-05,
      "epoch": 278.96,
      "step": 1115
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.03989477828145027,
      "learning_rate": 3.900343642611684e-05,
      "epoch": 280.0,
      "step": 1120
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.015496418811380863,
      "learning_rate": 3.871706758304697e-05,
      "epoch": 281.32,
      "step": 1125
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014337087981402874,
      "learning_rate": 3.8430698739977096e-05,
      "epoch": 282.64,
      "step": 1130
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013469228520989418,
      "learning_rate": 3.8144329896907216e-05,
      "epoch": 283.96,
      "step": 1135
    },
    {
      "loss": 0.0082,
      "grad_norm": 0.05482713133096695,
      "learning_rate": 3.785796105383734e-05,
      "epoch": 285.0,
      "step": 1140
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.014133197255432606,
      "learning_rate": 3.757159221076747e-05,
      "epoch": 286.32,
      "step": 1145
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013709329068660736,
      "learning_rate": 3.7285223367697595e-05,
      "epoch": 287.64,
      "step": 1150
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015217536129057407,
      "learning_rate": 3.699885452462772e-05,
      "epoch": 288.96,
      "step": 1155
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.05735105276107788,
      "learning_rate": 3.671248568155785e-05,
      "epoch": 290.0,
      "step": 1160
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.015944160521030426,
      "learning_rate": 3.6426116838487974e-05,
      "epoch": 291.32,
      "step": 1165
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.013248484581708908,
      "learning_rate": 3.61397479954181e-05,
      "epoch": 292.64,
      "step": 1170
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.015623633749783039,
      "learning_rate": 3.585337915234823e-05,
      "epoch": 293.96,
      "step": 1175
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.045159466564655304,
      "learning_rate": 3.5567010309278354e-05,
      "epoch": 295.0,
      "step": 1180
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.015232442878186703,
      "learning_rate": 3.528064146620848e-05,
      "epoch": 296.32,
      "step": 1185
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.013308205641806126,
      "learning_rate": 3.49942726231386e-05,
      "epoch": 297.64,
      "step": 1190
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.012608795426785946,
      "learning_rate": 3.4707903780068726e-05,
      "epoch": 298.96,
      "step": 1195
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.0456339456140995,
      "learning_rate": 3.442153493699885e-05,
      "epoch": 300.0,
      "step": 1200
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.013597261160612106,
      "learning_rate": 3.4135166093928986e-05,
      "epoch": 301.32,
      "step": 1205
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.016583947464823723,
      "learning_rate": 3.384879725085911e-05,
      "epoch": 302.64,
      "step": 1210
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.017008237540721893,
      "learning_rate": 3.356242840778923e-05,
      "epoch": 303.96,
      "step": 1215
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.04096021503210068,
      "learning_rate": 3.327605956471936e-05,
      "epoch": 305.0,
      "step": 1220
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.015342313796281815,
      "learning_rate": 3.2989690721649485e-05,
      "epoch": 306.32,
      "step": 1225
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.013403733260929585,
      "learning_rate": 3.270332187857961e-05,
      "epoch": 307.64,
      "step": 1230
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015681322664022446,
      "learning_rate": 3.241695303550974e-05,
      "epoch": 308.96,
      "step": 1235
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.04436800256371498,
      "learning_rate": 3.2130584192439865e-05,
      "epoch": 310.0,
      "step": 1240
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.012406053952872753,
      "learning_rate": 3.184421534936999e-05,
      "epoch": 311.32,
      "step": 1245
    },
    {
      "loss": 0.008,
      "grad_norm": 0.013188875280320644,
      "learning_rate": 3.155784650630012e-05,
      "epoch": 312.64,
      "step": 1250
    },
    {
      "loss": 0.008,
      "grad_norm": 0.012989694252610207,
      "learning_rate": 3.1271477663230244e-05,
      "epoch": 313.96,
      "step": 1255
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.03658341243863106,
      "learning_rate": 3.098510882016037e-05,
      "epoch": 315.0,
      "step": 1260
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.012549160048365593,
      "learning_rate": 3.06987399770905e-05,
      "epoch": 316.32,
      "step": 1265
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.014877067878842354,
      "learning_rate": 3.0412371134020617e-05,
      "epoch": 317.64,
      "step": 1270
    },
    {
      "loss": 0.0072,
      "grad_norm": 0.016601502895355225,
      "learning_rate": 3.0126002290950743e-05,
      "epoch": 318.96,
      "step": 1275
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.04804271459579468,
      "learning_rate": 2.983963344788087e-05,
      "epoch": 320.0,
      "step": 1280
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014096582308411598,
      "learning_rate": 2.9553264604811e-05,
      "epoch": 321.32,
      "step": 1285
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.015919970348477364,
      "learning_rate": 2.9266895761741126e-05,
      "epoch": 322.64,
      "step": 1290
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.015516753308475018,
      "learning_rate": 2.8980526918671253e-05,
      "epoch": 323.96,
      "step": 1295
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.045702189207077026,
      "learning_rate": 2.8694158075601372e-05,
      "epoch": 325.0,
      "step": 1300
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.014652375131845474,
      "learning_rate": 2.8407789232531502e-05,
      "epoch": 326.32,
      "step": 1305
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015444747172296047,
      "learning_rate": 2.812142038946163e-05,
      "epoch": 327.64,
      "step": 1310
    },
    {
      "loss": 0.008,
      "grad_norm": 0.01786893792450428,
      "learning_rate": 2.7835051546391755e-05,
      "epoch": 328.96,
      "step": 1315
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.036700889468193054,
      "learning_rate": 2.754868270332188e-05,
      "epoch": 330.0,
      "step": 1320
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015179547481238842,
      "learning_rate": 2.7262313860252005e-05,
      "epoch": 331.32,
      "step": 1325
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.013274077326059341,
      "learning_rate": 2.697594501718213e-05,
      "epoch": 332.64,
      "step": 1330
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014883404597640038,
      "learning_rate": 2.6689576174112258e-05,
      "epoch": 333.96,
      "step": 1335
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.03539450466632843,
      "learning_rate": 2.6403207331042384e-05,
      "epoch": 335.0,
      "step": 1340
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.01372632384300232,
      "learning_rate": 2.611683848797251e-05,
      "epoch": 336.32,
      "step": 1345
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.013471865095198154,
      "learning_rate": 2.5830469644902637e-05,
      "epoch": 337.64,
      "step": 1350
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014358654618263245,
      "learning_rate": 2.554410080183276e-05,
      "epoch": 338.96,
      "step": 1355
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.0419570691883564,
      "learning_rate": 2.5257731958762887e-05,
      "epoch": 340.0,
      "step": 1360
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013839066959917545,
      "learning_rate": 2.4971363115693013e-05,
      "epoch": 341.32,
      "step": 1365
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.01498588640242815,
      "learning_rate": 2.468499427262314e-05,
      "epoch": 342.64,
      "step": 1370
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.014100152999162674,
      "learning_rate": 2.4398625429553266e-05,
      "epoch": 343.96,
      "step": 1375
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.04237525910139084,
      "learning_rate": 2.4112256586483393e-05,
      "epoch": 345.0,
      "step": 1380
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.01373213343322277,
      "learning_rate": 2.3825887743413516e-05,
      "epoch": 346.32,
      "step": 1385
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.01432698592543602,
      "learning_rate": 2.3539518900343642e-05,
      "epoch": 347.64,
      "step": 1390
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.013069688342511654,
      "learning_rate": 2.3253150057273772e-05,
      "epoch": 348.96,
      "step": 1395
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.036869462579488754,
      "learning_rate": 2.2966781214203895e-05,
      "epoch": 350.0,
      "step": 1400
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.012810162268579006,
      "learning_rate": 2.268041237113402e-05,
      "epoch": 351.32,
      "step": 1405
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014795995317399502,
      "learning_rate": 2.2394043528064148e-05,
      "epoch": 352.64,
      "step": 1410
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.015119414776563644,
      "learning_rate": 2.210767468499427e-05,
      "epoch": 353.96,
      "step": 1415
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.051906559616327286,
      "learning_rate": 2.18213058419244e-05,
      "epoch": 355.0,
      "step": 1420
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.01508385594934225,
      "learning_rate": 2.1534936998854528e-05,
      "epoch": 356.32,
      "step": 1425
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014415285550057888,
      "learning_rate": 2.124856815578465e-05,
      "epoch": 357.64,
      "step": 1430
    },
    {
      "loss": 0.008,
      "grad_norm": 0.016101282089948654,
      "learning_rate": 2.0962199312714777e-05,
      "epoch": 358.96,
      "step": 1435
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.04253387823700905,
      "learning_rate": 2.0675830469644904e-05,
      "epoch": 360.0,
      "step": 1440
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014966638758778572,
      "learning_rate": 2.038946162657503e-05,
      "epoch": 361.32,
      "step": 1445
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015020391903817654,
      "learning_rate": 2.0103092783505157e-05,
      "epoch": 362.64,
      "step": 1450
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.013886959291994572,
      "learning_rate": 1.981672394043528e-05,
      "epoch": 363.96,
      "step": 1455
    },
    {
      "loss": 0.0086,
      "grad_norm": 0.05100937560200691,
      "learning_rate": 1.9530355097365406e-05,
      "epoch": 365.0,
      "step": 1460
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013956600800156593,
      "learning_rate": 1.9243986254295536e-05,
      "epoch": 366.32,
      "step": 1465
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014051297679543495,
      "learning_rate": 1.895761741122566e-05,
      "epoch": 367.64,
      "step": 1470
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.014400891028344631,
      "learning_rate": 1.8671248568155786e-05,
      "epoch": 368.96,
      "step": 1475
    },
    {
      "loss": 0.0072,
      "grad_norm": 0.045442789793014526,
      "learning_rate": 1.8384879725085912e-05,
      "epoch": 370.0,
      "step": 1480
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014202379621565342,
      "learning_rate": 1.809851088201604e-05,
      "epoch": 371.32,
      "step": 1485
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.014422114938497543,
      "learning_rate": 1.7812142038946165e-05,
      "epoch": 372.64,
      "step": 1490
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.013628999702632427,
      "learning_rate": 1.7525773195876288e-05,
      "epoch": 373.96,
      "step": 1495
    },
    {
      "loss": 0.0073,
      "grad_norm": 0.034053459763526917,
      "learning_rate": 1.7239404352806415e-05,
      "epoch": 375.0,
      "step": 1500
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.013981230556964874,
      "learning_rate": 1.695303550973654e-05,
      "epoch": 376.32,
      "step": 1505
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.012649217620491982,
      "learning_rate": 1.6666666666666667e-05,
      "epoch": 377.64,
      "step": 1510
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.01360363233834505,
      "learning_rate": 1.6380297823596794e-05,
      "epoch": 378.96,
      "step": 1515
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.05672921985387802,
      "learning_rate": 1.609392898052692e-05,
      "epoch": 380.0,
      "step": 1520
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013888300396502018,
      "learning_rate": 1.5807560137457044e-05,
      "epoch": 381.32,
      "step": 1525
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.013562100939452648,
      "learning_rate": 1.5521191294387173e-05,
      "epoch": 382.64,
      "step": 1530
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014399485662579536,
      "learning_rate": 1.5234822451317298e-05,
      "epoch": 383.96,
      "step": 1535
    },
    {
      "loss": 0.0087,
      "grad_norm": 0.05227126181125641,
      "learning_rate": 1.4948453608247423e-05,
      "epoch": 385.0,
      "step": 1540
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.0142129547894001,
      "learning_rate": 1.466208476517755e-05,
      "epoch": 386.32,
      "step": 1545
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.016521280631422997,
      "learning_rate": 1.4375715922107674e-05,
      "epoch": 387.64,
      "step": 1550
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.015063703060150146,
      "learning_rate": 1.40893470790378e-05,
      "epoch": 388.96,
      "step": 1555
    },
    {
      "loss": 0.008,
      "grad_norm": 0.05993450805544853,
      "learning_rate": 1.3802978235967929e-05,
      "epoch": 390.0,
      "step": 1560
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014749192632734776,
      "learning_rate": 1.3516609392898052e-05,
      "epoch": 391.32,
      "step": 1565
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014509294182062149,
      "learning_rate": 1.323024054982818e-05,
      "epoch": 392.64,
      "step": 1570
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013933155685663223,
      "learning_rate": 1.2943871706758307e-05,
      "epoch": 393.96,
      "step": 1575
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.046370115131139755,
      "learning_rate": 1.2657502863688431e-05,
      "epoch": 395.0,
      "step": 1580
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013361246325075626,
      "learning_rate": 1.2371134020618558e-05,
      "epoch": 396.32,
      "step": 1585
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.014140213839709759,
      "learning_rate": 1.2084765177548683e-05,
      "epoch": 397.64,
      "step": 1590
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.01335031259804964,
      "learning_rate": 1.1798396334478809e-05,
      "epoch": 398.96,
      "step": 1595
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.04303241893649101,
      "learning_rate": 1.1512027491408934e-05,
      "epoch": 400.0,
      "step": 1600
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.01495408732444048,
      "learning_rate": 1.1225658648339062e-05,
      "epoch": 401.32,
      "step": 1605
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014716042205691338,
      "learning_rate": 1.0939289805269187e-05,
      "epoch": 402.64,
      "step": 1610
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.013150262646377087,
      "learning_rate": 1.0652920962199313e-05,
      "epoch": 403.96,
      "step": 1615
    },
    {
      "loss": 0.0081,
      "grad_norm": 0.05694189667701721,
      "learning_rate": 1.036655211912944e-05,
      "epoch": 405.0,
      "step": 1620
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.014942456036806107,
      "learning_rate": 1.0080183276059566e-05,
      "epoch": 406.32,
      "step": 1625
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.014433029107749462,
      "learning_rate": 9.793814432989691e-06,
      "epoch": 407.64,
      "step": 1630
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.013919273391366005,
      "learning_rate": 9.507445589919818e-06,
      "epoch": 408.96,
      "step": 1635
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.05335159972310066,
      "learning_rate": 9.221076746849944e-06,
      "epoch": 410.0,
      "step": 1640
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.012211896479129791,
      "learning_rate": 8.934707903780069e-06,
      "epoch": 411.32,
      "step": 1645
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.015226286835968494,
      "learning_rate": 8.648339060710195e-06,
      "epoch": 412.64,
      "step": 1650
    },
    {
      "loss": 0.0073,
      "grad_norm": 0.013444135896861553,
      "learning_rate": 8.36197021764032e-06,
      "epoch": 413.96,
      "step": 1655
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.04133617505431175,
      "learning_rate": 8.075601374570448e-06,
      "epoch": 415.0,
      "step": 1660
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.012831657193601131,
      "learning_rate": 7.789232531500573e-06,
      "epoch": 416.32,
      "step": 1665
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.013414805755019188,
      "learning_rate": 7.502863688430699e-06,
      "epoch": 417.64,
      "step": 1670
    },
    {
      "loss": 0.0073,
      "grad_norm": 0.013841026462614536,
      "learning_rate": 7.216494845360824e-06,
      "epoch": 418.96,
      "step": 1675
    },
    {
      "loss": 0.0079,
      "grad_norm": 0.05774722248315811,
      "learning_rate": 6.930126002290952e-06,
      "epoch": 420.0,
      "step": 1680
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.016231786459684372,
      "learning_rate": 6.643757159221077e-06,
      "epoch": 421.32,
      "step": 1685
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.013711273670196533,
      "learning_rate": 6.357388316151203e-06,
      "epoch": 422.64,
      "step": 1690
    },
    {
      "loss": 0.008,
      "grad_norm": 0.015432053245604038,
      "learning_rate": 6.071019473081329e-06,
      "epoch": 423.96,
      "step": 1695
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.04593059420585632,
      "learning_rate": 5.784650630011455e-06,
      "epoch": 425.0,
      "step": 1700
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.013549490831792355,
      "learning_rate": 5.498281786941581e-06,
      "epoch": 426.32,
      "step": 1705
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.015252575278282166,
      "learning_rate": 5.211912943871707e-06,
      "epoch": 427.64,
      "step": 1710
    },
    {
      "loss": 0.0077,
      "grad_norm": 0.015331484377384186,
      "learning_rate": 4.925544100801833e-06,
      "epoch": 428.96,
      "step": 1715
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.053916484117507935,
      "learning_rate": 4.639175257731959e-06,
      "epoch": 430.0,
      "step": 1720
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.014025518670678139,
      "learning_rate": 4.352806414662085e-06,
      "epoch": 431.32,
      "step": 1725
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014415928162634373,
      "learning_rate": 4.066437571592211e-06,
      "epoch": 432.64,
      "step": 1730
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.016451990231871605,
      "learning_rate": 3.7800687285223365e-06,
      "epoch": 433.96,
      "step": 1735
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.04959571361541748,
      "learning_rate": 3.493699885452463e-06,
      "epoch": 435.0,
      "step": 1740
    },
    {
      "loss": 0.008,
      "grad_norm": 0.014649420976638794,
      "learning_rate": 3.2073310423825886e-06,
      "epoch": 436.32,
      "step": 1745
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.015780117362737656,
      "learning_rate": 2.920962199312715e-06,
      "epoch": 437.64,
      "step": 1750
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.014374195598065853,
      "learning_rate": 2.6345933562428407e-06,
      "epoch": 438.96,
      "step": 1755
    },
    {
      "loss": 0.007,
      "grad_norm": 0.04255754500627518,
      "learning_rate": 2.3482245131729668e-06,
      "epoch": 440.0,
      "step": 1760
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014756709337234497,
      "learning_rate": 2.061855670103093e-06,
      "epoch": 441.32,
      "step": 1765
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.014627397991716862,
      "learning_rate": 1.7754868270332189e-06,
      "epoch": 442.64,
      "step": 1770
    },
    {
      "loss": 0.0073,
      "grad_norm": 0.013303273357450962,
      "learning_rate": 1.4891179839633447e-06,
      "epoch": 443.96,
      "step": 1775
    },
    {
      "loss": 0.0083,
      "grad_norm": 0.055010970681905746,
      "learning_rate": 1.202749140893471e-06,
      "epoch": 445.0,
      "step": 1780
    },
    {
      "loss": 0.0078,
      "grad_norm": 0.01359732635319233,
      "learning_rate": 9.163802978235968e-07,
      "epoch": 446.32,
      "step": 1785
    },
    {
      "loss": 0.0074,
      "grad_norm": 0.012304678559303284,
      "learning_rate": 6.300114547537229e-07,
      "epoch": 447.64,
      "step": 1790
    },
    {
      "loss": 0.0075,
      "grad_norm": 0.014455261640250683,
      "learning_rate": 3.436426116838488e-07,
      "epoch": 448.96,
      "step": 1795
    },
    {
      "loss": 0.0076,
      "grad_norm": 0.04836144670844078,
      "learning_rate": 5.72737686139748e-08,
      "epoch": 450.0,
      "step": 1800
    },
    {
      "train_runtime": 30699.1941,
      "train_samples_per_second": 1.876,
      "train_steps_per_second": 0.059,
      "total_flos": 1.5308141101056e+18,
      "train_loss": 0.04049773897975683,
      "epoch": 450.0,
      "step": 1800
    }
  ],
  "training_args": {
    "output_dir": "/hkfs/work/workspace/scratch/tum_fmp0582-dndworkspace/自己训练lora/train_lora/outputs/ToT-Math-V1/group_02/checkpoints",
    "overwrite_output_dir": false,
    "do_train": false,
    "do_eval": false,
    "do_predict": false,
    "eval_strategy": "no",
    "prediction_loss_only": false,
    "per_device_train_batch_size": 2,
    "per_device_eval_batch_size": 8,
    "per_gpu_train_batch_size": null,
    "per_gpu_eval_batch_size": null,
    "gradient_accumulation_steps": 16,
    "eval_accumulation_steps": null,
    "eval_delay": 0,
    "torch_empty_cache_steps": null,
    "learning_rate": 0.0001,
    "weight_decay": 0.01,
    "adam_beta1": 0.9,
    "adam_beta2": 0.999,
    "adam_epsilon": 1e-08,
    "max_grad_norm": 1.0,
    "num_train_epochs": 12,
    "max_steps": 1800,
    "lr_scheduler_type": "linear",
    "lr_scheduler_kwargs": {},
    "warmup_ratio": 0.03,
    "warmup_steps": 0,
    "log_level": "passive",
    "log_level_replica": "warning",
    "log_on_each_node": true,
    "logging_dir": "/hkfs/work/workspace/scratch/tum_fmp0582-dndworkspace/自己训练lora/train_lora/logs/ToT-Math-V1/group_02",
    "logging_strategy": "steps",
    "logging_first_step": true,
    "logging_steps": 5,
    "logging_nan_inf_filter": true,
    "save_strategy": "steps",
    "save_steps": 300,
    "save_total_limit": 6,
    "save_safetensors": true,
    "save_on_each_node": false,
    "save_only_model": false,
    "restore_callback_states_from_checkpoint": false,
    "no_cuda": false,
    "use_cpu": false,
    "use_mps_device": false,
    "seed": 42,
    "data_seed": null,
    "jit_mode_eval": false,
    "bf16": true,
    "fp16": false,
    "fp16_opt_level": "O1",
    "half_precision_backend": "auto",
    "bf16_full_eval": false,
    "fp16_full_eval": false,
    "tf32": null,
    "local_rank": 0,
    "ddp_backend": null,
    "tpu_num_cores": null,
    "tpu_metrics_debug": false,
    "debug": [],
    "dataloader_drop_last": false,
    "eval_steps": null,
    "dataloader_num_workers": 0,
    "dataloader_prefetch_factor": null,
    "past_index": -1,
    "run_name": null,
    "disable_tqdm": false,
    "remove_unused_columns": true,
    "label_names": null,
    "load_best_model_at_end": false,
    "metric_for_best_model": null,
    "greater_is_better": null,
    "ignore_data_skip": false,
    "fsdp": [],
    "fsdp_min_num_params": 0,
    "fsdp_config": {
      "min_num_params": 0,
      "xla": false,
      "xla_fsdp_v2": false,
      "xla_fsdp_grad_ckpt": false
    },
    "fsdp_transformer_layer_cls_to_wrap": null,
    "accelerator_config": {
      "split_batches": false,
      "dispatch_batches": null,
      "even_batches": true,
      "use_seedable_sampler": true,
      "non_blocking": false,
      "gradient_accumulation_kwargs": null
    },
    "parallelism_config": null,
    "deepspeed": null,
    "label_smoothing_factor": 0.0,
    "optim": "adamw_torch",
    "optim_args": null,
    "adafactor": false,
    "group_by_length": false,
    "length_column_name": "length",
    "report_to": [],
    "project": "huggingface",
    "trackio_space_id": "trackio",
    "ddp_find_unused_parameters": null,
    "ddp_bucket_cap_mb": null,
    "ddp_broadcast_buffers": null,
    "dataloader_pin_memory": true,
    "dataloader_persistent_workers": false,
    "skip_memory_metrics": true,
    "use_legacy_prediction_loop": false,
    "push_to_hub": false,
    "resume_from_checkpoint": null,
    "hub_model_id": null,
    "hub_strategy": "every_save",
    "hub_token": "<HUB_TOKEN>",
    "hub_private_repo": null,
    "hub_always_push": false,
    "hub_revision": null,
    "gradient_checkpointing": true,
    "gradient_checkpointing_kwargs": null,
    "include_inputs_for_metrics": false,
    "include_for_metrics": [],
    "eval_do_concat_batches": true,
    "fp16_backend": "auto",
    "push_to_hub_model_id": null,
    "push_to_hub_organization": null,
    "push_to_hub_token": "<PUSH_TO_HUB_TOKEN>",
    "mp_parameters": "",
    "auto_find_batch_size": false,
    "full_determinism": false,
    "torchdynamo": null,
    "ray_scope": "last",
    "ddp_timeout": 1800,
    "torch_compile": false,
    "torch_compile_backend": null,
    "torch_compile_mode": null,
    "include_tokens_per_second": false,
    "include_num_input_tokens_seen": "no",
    "neftune_noise_alpha": null,
    "optim_target_modules": null,
    "batch_eval_metrics": false,
    "eval_on_start": false,
    "use_liger_kernel": false,
    "liger_kernel_config": null,
    "eval_use_gather_object": false,
    "average_tokens_across_devices": true
  },
  "lora_config": {
    "r": 64,
    "alpha": 128,
    "dropout": 0.05,
    "target_modules": [
      "q_proj",
      "k_proj",
      "v_proj",
      "o_proj",
      "gate_proj",
      "up_proj",
      "down_proj"
    ]
  },
  "effective_batch_size": 32,
  "world_size": 1,
  "git_commit": ""
}