{
  "best_metric": 91.75966152710339,
  "best_model_checkpoint": "/scratch/mrahma45/pixel/finetuned_models/mbert/mbert-base-finetuned-parsing-ud-Hindi-HDTB/checkpoint-6500",
  "epoch": 21.634615384615383,
  "global_step": 9000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.24,
      "learning_rate": 7.840000000000001e-05,
      "loss": 2.8321,
      "step": 100
    },
    {
      "epoch": 0.48,
      "learning_rate": 7.947382550335571e-05,
      "loss": 0.7588,
      "step": 200
    },
    {
      "epoch": 0.72,
      "learning_rate": 7.893691275167786e-05,
      "loss": 0.607,
      "step": 300
    },
    {
      "epoch": 0.96,
      "learning_rate": 7.840000000000001e-05,
      "loss": 0.5116,
      "step": 400
    },
    {
      "epoch": 1.2,
      "learning_rate": 7.786308724832216e-05,
      "loss": 0.3944,
      "step": 500
    },
    {
      "epoch": 1.2,
      "eval_las": 89.23531249112644,
      "eval_loss": 0.415132611989975,
      "eval_runtime": 11.8077,
      "eval_samples_per_second": 140.501,
      "eval_steps_per_second": 17.616,
      "eval_uas": 93.45202601016554,
      "step": 500
    },
    {
      "epoch": 1.44,
      "learning_rate": 7.73261744966443e-05,
      "loss": 0.3888,
      "step": 600
    },
    {
      "epoch": 1.68,
      "learning_rate": 7.678926174496645e-05,
      "loss": 0.3689,
      "step": 700
    },
    {
      "epoch": 1.92,
      "learning_rate": 7.62523489932886e-05,
      "loss": 0.3628,
      "step": 800
    },
    {
      "epoch": 2.16,
      "learning_rate": 7.571543624161075e-05,
      "loss": 0.291,
      "step": 900
    },
    {
      "epoch": 2.4,
      "learning_rate": 7.51785234899329e-05,
      "loss": 0.265,
      "step": 1000
    },
    {
      "epoch": 2.4,
      "eval_las": 90.46199278757419,
      "eval_loss": 0.37675201892852783,
      "eval_runtime": 11.7912,
      "eval_samples_per_second": 140.698,
      "eval_steps_per_second": 17.64,
      "eval_uas": 94.2726524121873,
      "step": 1000
    },
    {
      "epoch": 2.64,
      "learning_rate": 7.464161073825505e-05,
      "loss": 0.2676,
      "step": 1100
    },
    {
      "epoch": 2.88,
      "learning_rate": 7.410469798657718e-05,
      "loss": 0.2719,
      "step": 1200
    },
    {
      "epoch": 3.12,
      "learning_rate": 7.356778523489933e-05,
      "loss": 0.2325,
      "step": 1300
    },
    {
      "epoch": 3.37,
      "learning_rate": 7.303087248322148e-05,
      "loss": 0.2006,
      "step": 1400
    },
    {
      "epoch": 3.61,
      "learning_rate": 7.249395973154363e-05,
      "loss": 0.2103,
      "step": 1500
    },
    {
      "epoch": 3.61,
      "eval_las": 91.04409802084221,
      "eval_loss": 0.3845142126083374,
      "eval_runtime": 11.7995,
      "eval_samples_per_second": 140.6,
      "eval_steps_per_second": 17.628,
      "eval_uas": 94.56512479768294,
      "step": 1500
    },
    {
      "epoch": 3.85,
      "learning_rate": 7.195704697986577e-05,
      "loss": 0.2208,
      "step": 1600
    },
    {
      "epoch": 4.09,
      "learning_rate": 7.142013422818792e-05,
      "loss": 0.1948,
      "step": 1700
    },
    {
      "epoch": 4.33,
      "learning_rate": 7.088322147651007e-05,
      "loss": 0.1553,
      "step": 1800
    },
    {
      "epoch": 4.57,
      "learning_rate": 7.034630872483222e-05,
      "loss": 0.1593,
      "step": 1900
    },
    {
      "epoch": 4.81,
      "learning_rate": 6.980939597315437e-05,
      "loss": 0.1655,
      "step": 2000
    },
    {
      "epoch": 4.81,
      "eval_las": 91.03557940767243,
      "eval_loss": 0.3911457359790802,
      "eval_runtime": 11.7868,
      "eval_samples_per_second": 140.751,
      "eval_steps_per_second": 17.647,
      "eval_uas": 94.65882954255048,
      "step": 2000
    },
    {
      "epoch": 5.05,
      "learning_rate": 6.927248322147651e-05,
      "loss": 0.1562,
      "step": 2100
    },
    {
      "epoch": 5.29,
      "learning_rate": 6.873557046979866e-05,
      "loss": 0.1157,
      "step": 2200
    },
    {
      "epoch": 5.53,
      "learning_rate": 6.819865771812081e-05,
      "loss": 0.1221,
      "step": 2300
    },
    {
      "epoch": 5.77,
      "learning_rate": 6.766174496644296e-05,
      "loss": 0.1336,
      "step": 2400
    },
    {
      "epoch": 6.01,
      "learning_rate": 6.712483221476511e-05,
      "loss": 0.1319,
      "step": 2500
    },
    {
      "epoch": 6.01,
      "eval_las": 90.95323281369792,
      "eval_loss": 0.44331711530685425,
      "eval_runtime": 11.7942,
      "eval_samples_per_second": 140.663,
      "eval_steps_per_second": 17.636,
      "eval_uas": 94.61055740125508,
      "step": 2500
    },
    {
      "epoch": 6.25,
      "learning_rate": 6.658791946308726e-05,
      "loss": 0.0893,
      "step": 2600
    },
    {
      "epoch": 6.49,
      "learning_rate": 6.60510067114094e-05,
      "loss": 0.0981,
      "step": 2700
    },
    {
      "epoch": 6.73,
      "learning_rate": 6.551409395973155e-05,
      "loss": 0.099,
      "step": 2800
    },
    {
      "epoch": 6.97,
      "learning_rate": 6.49771812080537e-05,
      "loss": 0.1028,
      "step": 2900
    },
    {
      "epoch": 7.21,
      "learning_rate": 6.444026845637585e-05,
      "loss": 0.0774,
      "step": 3000
    },
    {
      "epoch": 7.21,
      "eval_las": 91.02138171905614,
      "eval_loss": 0.534425675868988,
      "eval_runtime": 11.7987,
      "eval_samples_per_second": 140.609,
      "eval_steps_per_second": 17.629,
      "eval_uas": 94.7184598347389,
      "step": 3000
    },
    {
      "epoch": 7.45,
      "learning_rate": 6.3903355704698e-05,
      "loss": 0.0774,
      "step": 3100
    },
    {
      "epoch": 7.69,
      "learning_rate": 6.336644295302015e-05,
      "loss": 0.0815,
      "step": 3200
    },
    {
      "epoch": 7.93,
      "learning_rate": 6.28295302013423e-05,
      "loss": 0.0855,
      "step": 3300
    },
    {
      "epoch": 8.17,
      "learning_rate": 6.229261744966444e-05,
      "loss": 0.0633,
      "step": 3400
    },
    {
      "epoch": 8.41,
      "learning_rate": 6.175570469798658e-05,
      "loss": 0.0655,
      "step": 3500
    },
    {
      "epoch": 8.41,
      "eval_las": 91.10088877530738,
      "eval_loss": 0.5579993724822998,
      "eval_runtime": 11.8126,
      "eval_samples_per_second": 140.444,
      "eval_steps_per_second": 17.608,
      "eval_uas": 94.6304341653179,
      "step": 3500
    },
    {
      "epoch": 8.65,
      "learning_rate": 6.121879194630873e-05,
      "loss": 0.0654,
      "step": 3600
    },
    {
      "epoch": 8.89,
      "learning_rate": 6.068187919463087e-05,
      "loss": 0.0692,
      "step": 3700
    },
    {
      "epoch": 9.13,
      "learning_rate": 6.014496644295302e-05,
      "loss": 0.0608,
      "step": 3800
    },
    {
      "epoch": 9.38,
      "learning_rate": 5.9613422818791955e-05,
      "loss": 0.0529,
      "step": 3900
    },
    {
      "epoch": 9.62,
      "learning_rate": 5.90765100671141e-05,
      "loss": 0.0568,
      "step": 4000
    },
    {
      "epoch": 9.62,
      "eval_las": 91.29397734048896,
      "eval_loss": 0.5721077919006348,
      "eval_runtime": 11.8178,
      "eval_samples_per_second": 140.382,
      "eval_steps_per_second": 17.601,
      "eval_uas": 94.87179487179486,
      "step": 4000
    },
    {
      "epoch": 9.86,
      "learning_rate": 5.853959731543625e-05,
      "loss": 0.0554,
      "step": 4100
    },
    {
      "epoch": 10.1,
      "learning_rate": 5.80026845637584e-05,
      "loss": 0.0517,
      "step": 4200
    },
    {
      "epoch": 10.34,
      "learning_rate": 5.7465771812080534e-05,
      "loss": 0.0441,
      "step": 4300
    },
    {
      "epoch": 10.58,
      "learning_rate": 5.692885906040268e-05,
      "loss": 0.0471,
      "step": 4400
    },
    {
      "epoch": 10.82,
      "learning_rate": 5.639194630872483e-05,
      "loss": 0.0462,
      "step": 4500
    },
    {
      "epoch": 10.82,
      "eval_las": 91.28261918959593,
      "eval_loss": 0.6217162609100342,
      "eval_runtime": 11.7974,
      "eval_samples_per_second": 140.624,
      "eval_steps_per_second": 17.631,
      "eval_uas": 94.88599256041117,
      "step": 4500
    },
    {
      "epoch": 11.06,
      "learning_rate": 5.5855033557046986e-05,
      "loss": 0.0483,
      "step": 4600
    },
    {
      "epoch": 11.3,
      "learning_rate": 5.5318120805369134e-05,
      "loss": 0.0393,
      "step": 4700
    },
    {
      "epoch": 11.54,
      "learning_rate": 5.478120805369128e-05,
      "loss": 0.0402,
      "step": 4800
    },
    {
      "epoch": 11.78,
      "learning_rate": 5.424429530201343e-05,
      "loss": 0.0424,
      "step": 4900
    },
    {
      "epoch": 12.02,
      "learning_rate": 5.370738255033558e-05,
      "loss": 0.0418,
      "step": 5000
    },
    {
      "epoch": 12.02,
      "eval_las": 91.32237271772155,
      "eval_loss": 0.6571192145347595,
      "eval_runtime": 11.8066,
      "eval_samples_per_second": 140.515,
      "eval_steps_per_second": 17.617,
      "eval_uas": 94.87747394724138,
      "step": 5000
    },
    {
      "epoch": 12.26,
      "learning_rate": 5.317046979865773e-05,
      "loss": 0.0339,
      "step": 5100
    },
    {
      "epoch": 12.5,
      "learning_rate": 5.2633557046979875e-05,
      "loss": 0.0361,
      "step": 5200
    },
    {
      "epoch": 12.74,
      "learning_rate": 5.2096644295302024e-05,
      "loss": 0.0347,
      "step": 5300
    },
    {
      "epoch": 12.98,
      "learning_rate": 5.155973154362417e-05,
      "loss": 0.0374,
      "step": 5400
    },
    {
      "epoch": 13.22,
      "learning_rate": 5.1022818791946307e-05,
      "loss": 0.0304,
      "step": 5500
    },
    {
      "epoch": 13.22,
      "eval_las": 91.43879376437516,
      "eval_loss": 0.6834462881088257,
      "eval_runtime": 11.776,
      "eval_samples_per_second": 140.879,
      "eval_steps_per_second": 17.663,
      "eval_uas": 95.01661129568106,
      "step": 5500
    },
    {
      "epoch": 13.46,
      "learning_rate": 5.0485906040268455e-05,
      "loss": 0.0299,
      "step": 5600
    },
    {
      "epoch": 13.7,
      "learning_rate": 4.99489932885906e-05,
      "loss": 0.0335,
      "step": 5700
    },
    {
      "epoch": 13.94,
      "learning_rate": 4.941208053691275e-05,
      "loss": 0.0331,
      "step": 5800
    },
    {
      "epoch": 14.18,
      "learning_rate": 4.88751677852349e-05,
      "loss": 0.026,
      "step": 5900
    },
    {
      "epoch": 14.42,
      "learning_rate": 4.833825503355705e-05,
      "loss": 0.0255,
      "step": 6000
    },
    {
      "epoch": 14.42,
      "eval_las": 91.52965897151944,
      "eval_loss": 0.7086959481239319,
      "eval_runtime": 11.7728,
      "eval_samples_per_second": 140.918,
      "eval_steps_per_second": 17.668,
      "eval_uas": 95.10179742737883,
      "step": 6000
    },
    {
      "epoch": 14.66,
      "learning_rate": 4.7801342281879196e-05,
      "loss": 0.0273,
      "step": 6100
    },
    {
      "epoch": 14.9,
      "learning_rate": 4.726442953020135e-05,
      "loss": 0.0286,
      "step": 6200
    },
    {
      "epoch": 15.14,
      "learning_rate": 4.67275167785235e-05,
      "loss": 0.0219,
      "step": 6300
    },
    {
      "epoch": 15.38,
      "learning_rate": 4.619060402684565e-05,
      "loss": 0.0216,
      "step": 6400
    },
    {
      "epoch": 15.62,
      "learning_rate": 4.565906040268457e-05,
      "loss": 0.0254,
      "step": 6500
    },
    {
      "epoch": 15.62,
      "eval_las": 91.75966152710339,
      "eval_loss": 0.7071714401245117,
      "eval_runtime": 11.7877,
      "eval_samples_per_second": 140.74,
      "eval_steps_per_second": 17.646,
      "eval_uas": 95.23525570037198,
      "step": 6500
    },
    {
      "epoch": 15.87,
      "learning_rate": 4.5122147651006716e-05,
      "loss": 0.0242,
      "step": 6600
    },
    {
      "epoch": 16.11,
      "learning_rate": 4.4585234899328864e-05,
      "loss": 0.023,
      "step": 6700
    },
    {
      "epoch": 16.35,
      "learning_rate": 4.404832214765101e-05,
      "loss": 0.0222,
      "step": 6800
    },
    {
      "epoch": 16.59,
      "learning_rate": 4.351140939597316e-05,
      "loss": 0.0198,
      "step": 6900
    },
    {
      "epoch": 16.83,
      "learning_rate": 4.297449664429531e-05,
      "loss": 0.0211,
      "step": 7000
    },
    {
      "epoch": 16.83,
      "eval_las": 91.60632649004742,
      "eval_loss": 0.7650447487831116,
      "eval_runtime": 11.8115,
      "eval_samples_per_second": 140.456,
      "eval_steps_per_second": 17.61,
      "eval_uas": 95.07908112559275,
      "step": 7000
    },
    {
      "epoch": 17.07,
      "learning_rate": 4.243758389261746e-05,
      "loss": 0.0202,
      "step": 7100
    },
    {
      "epoch": 17.31,
      "learning_rate": 4.1900671140939605e-05,
      "loss": 0.0165,
      "step": 7200
    },
    {
      "epoch": 17.55,
      "learning_rate": 4.1363758389261754e-05,
      "loss": 0.021,
      "step": 7300
    },
    {
      "epoch": 17.79,
      "learning_rate": 4.0826845637583895e-05,
      "loss": 0.0175,
      "step": 7400
    },
    {
      "epoch": 18.03,
      "learning_rate": 4.028993288590604e-05,
      "loss": 0.019,
      "step": 7500
    },
    {
      "epoch": 18.03,
      "eval_las": 91.65743816906607,
      "eval_loss": 0.7897760272026062,
      "eval_runtime": 11.7771,
      "eval_samples_per_second": 140.867,
      "eval_steps_per_second": 17.661,
      "eval_uas": 95.08759973876253,
      "step": 7500
    },
    {
      "epoch": 18.27,
      "learning_rate": 3.975302013422819e-05,
      "loss": 0.015,
      "step": 7600
    },
    {
      "epoch": 18.51,
      "learning_rate": 3.921610738255034e-05,
      "loss": 0.0166,
      "step": 7700
    },
    {
      "epoch": 18.75,
      "learning_rate": 3.867919463087249e-05,
      "loss": 0.0165,
      "step": 7800
    },
    {
      "epoch": 18.99,
      "learning_rate": 3.8142281879194636e-05,
      "loss": 0.0156,
      "step": 7900
    },
    {
      "epoch": 19.23,
      "learning_rate": 3.7605369127516784e-05,
      "loss": 0.0125,
      "step": 8000
    },
    {
      "epoch": 19.23,
      "eval_las": 91.5183008206264,
      "eval_loss": 0.7997450828552246,
      "eval_runtime": 11.8137,
      "eval_samples_per_second": 140.431,
      "eval_steps_per_second": 17.607,
      "eval_uas": 94.9995740693415,
      "step": 8000
    },
    {
      "epoch": 19.47,
      "learning_rate": 3.706845637583893e-05,
      "loss": 0.0155,
      "step": 8100
    },
    {
      "epoch": 19.71,
      "learning_rate": 3.653154362416108e-05,
      "loss": 0.0155,
      "step": 8200
    },
    {
      "epoch": 19.95,
      "learning_rate": 3.599463087248322e-05,
      "loss": 0.0142,
      "step": 8300
    },
    {
      "epoch": 20.19,
      "learning_rate": 3.545771812080537e-05,
      "loss": 0.0096,
      "step": 8400
    },
    {
      "epoch": 20.43,
      "learning_rate": 3.492080536912752e-05,
      "loss": 0.0132,
      "step": 8500
    },
    {
      "epoch": 20.43,
      "eval_las": 91.67731493312888,
      "eval_loss": 0.8170235753059387,
      "eval_runtime": 11.7652,
      "eval_samples_per_second": 141.009,
      "eval_steps_per_second": 17.679,
      "eval_uas": 95.12735326688815,
      "step": 8500
    },
    {
      "epoch": 20.67,
      "learning_rate": 3.438389261744967e-05,
      "loss": 0.0122,
      "step": 8600
    },
    {
      "epoch": 20.91,
      "learning_rate": 3.3846979865771815e-05,
      "loss": 0.0116,
      "step": 8700
    },
    {
      "epoch": 21.15,
      "learning_rate": 3.3310067114093964e-05,
      "loss": 0.0114,
      "step": 8800
    },
    {
      "epoch": 21.39,
      "learning_rate": 3.277315436241611e-05,
      "loss": 0.0089,
      "step": 8900
    },
    {
      "epoch": 21.63,
      "learning_rate": 3.223624161073826e-05,
      "loss": 0.0115,
      "step": 9000
    },
    {
      "epoch": 21.63,
      "eval_las": 91.60348695232416,
      "eval_loss": 0.8960289359092712,
      "eval_runtime": 12.1056,
      "eval_samples_per_second": 137.044,
      "eval_steps_per_second": 17.182,
      "eval_uas": 95.08759973876253,
      "step": 9000
    },
    {
      "epoch": 21.63,
      "step": 9000,
      "total_flos": 4.80612874451927e+16,
      "train_loss": 0.13298969575431613,
      "train_runtime": 5206.9637,
      "train_samples_per_second": 92.184,
      "train_steps_per_second": 2.881
    }
  ],
  "max_steps": 15000,
  "num_train_epochs": 37,
  "total_flos": 4.80612874451927e+16,
  "trial_name": null,
  "trial_params": null
}