Training in progress, step 1000, checkpoint

last-checkpoint/config.json

@@ -8,184 +8,184 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "Water Supply",
-    "1": "Sexual Orientation and Gender Identity",
-    "2": "Investment Climate",
-    "3": "Environment",
-    "4": "Water Resources Management",
-    "5": "Labor Markets",
-    "6": "Governance",
-    "7": "Competition Policy",
-    "8": "Long-Term Finance",
-    "9": "Food Security Update",
-    "10": "Social Protection",
-    "11": "Inequality and Shared Prosperity",
-    "12": "Infectious diseases and Vaccines",
-    "13": "Jobs and Development",
-    "14": "Energy",
-    "15": "Education and Technology",
-    "16": "Debt Relief",
-    "17": "Financial Sector",
-    "18": "Measuring Poverty",
-    "19": "Education",
-    "20": "Water",
-    "21": "Procurement for Development",
-    "22": "Livestock and Sustainability",
-    "23": "Disaster Risk Management",
-    "24": "Financial Inclusion",
-    "25": "Trade Facilitation and Logistics",
-    "26": "Financial Integrity",
-    "27": "Health",
-    "28": "Regional Integration",
-    "29": "Forests and Landscapes",
-    "30": "Jobs & Development",
-    "31": "Payment Systems",
-    "32": "Agribusiness and Value Chains",
-    "33": "Disability Inclusion",
-    "34": "Marine Plastic Pollution",
-    "35": "Global Financing Facility for Women, Children and Adolescents",
-    "36": "Skills Development",
-    "37": "Sustainable Infrastructure Finance",
-    "38": "Land",
-    "39": "Sustainable Communities",
-    "40": "Higher Education",
-    "41": "Climate Change",
-    "42": "Small and Medium Enterprises Finance",
-    "43": "Teachers",
-    "44": "Migration",
-    "45": "Debt",
-    "46": "Urban Development",
-    "47": "Mining Investment and Governance Review",
-    "48": "Innovation and Entrepreneurship",
-    "49": "Natural Capital",
-    "50": "Transport",
-    "51": "Gender",
-    "52": "Safety Nets and Cash Transfers",
-    "53": "Inclusive Cities",
-    "54": "Extractive Industries",
-    "55": "Global Value Chains",
-    "56": "Agriculture and Food",
-    "57": "Universal Health Coverage",
-    "58": "Sanitation",
-    "59": "Gas Flaring Reduction",
-    "60": "COVID-19 Hub",
-    "61": "Girls' Education",
-    "62": "Macroeconomics",
-    "63": "Trade",
-    "64": "Pandemic Preparedness and COVID-19",
-    "65": "Infrastructure",
-    "66": "Oceans, Fisheries, and Coastal Economies",
-    "67": "Social Sustainability and Inclusion",
-    "68": "Community-Driven Development",
-    "69": "Credit Infrastructure",
-    "70": "Pollution",
-    "71": "Pensions",
-    "72": "Financial Stability",
-    "73": "One Health",
-    "74": "Indigenous Peoples",
-    "75": "Digital Development",
-    "76": "Climate-Smart Agriculture",
-    "77": "Food System Jobs",
-    "78": "Early Childhood Development",
-    "79": "Biodiversity",
-    "80": "Poverty",
-    "81": "Fragility, Conflict, and Violence",
-    "82": "Competitiveness",
-    "83": "Social Inclusion",
-    "84": "Nutrition",
-    "85": "Taxes and Government Revenue",
-    "86": "Water in Agriculture"
+    "0": "Global Financing Facility for Women, Children and Adolescents",
+    "1": "Competitiveness",
+    "2": "Sustainable Communities",
+    "3": "Education",
+    "4": "Natural Capital",
+    "5": "Regional Integration",
+    "6": "Jobs and Development",
+    "7": "Innovation and Entrepreneurship",
+    "8": "Inclusive Cities",
+    "9": "Jobs & Development",
+    "10": "Transport",
+    "11": "Sustainable Infrastructure Finance",
+    "12": "Pandemic Preparedness and COVID-19",
+    "13": "Taxes and Government Revenue",
+    "14": "Labor Markets",
+    "15": "Measuring Poverty",
+    "16": "Food Security Update",
+    "17": "Disability Inclusion",
+    "18": "Agriculture and Food",
+    "19": "Trade Facilitation and Logistics",
+    "20": "Social Protection",
+    "21": "One Health",
+    "22": "Biodiversity",
+    "23": "Social Inclusion",
+    "24": "Credit Infrastructure",
+    "25": "Water Supply",
+    "26": "Early Childhood Development",
+    "27": "Food System Jobs",
+    "28": "Migration",
+    "29": "Indigenous Peoples",
+    "30": "Universal Health Coverage",
+    "31": "Financial Sector",
+    "32": "Procurement for Development",
+    "33": "Inequality and Shared Prosperity",
+    "34": "COVID-19 Hub",
+    "35": "Poverty",
+    "36": "Financial Stability",
+    "37": "Digital Development",
+    "38": "Long-Term Finance",
+    "39": "Gas Flaring Reduction",
+    "40": "Mining Investment and Governance Review",
+    "41": "Small and Medium Enterprises Finance",
+    "42": "Infrastructure",
+    "43": "Health",
+    "44": "Sexual Orientation and Gender Identity",
+    "45": "Nutrition",
+    "46": "Financial Inclusion",
+    "47": "Fragility, Conflict, and Violence",
+    "48": "Debt Relief",
+    "49": "Disaster Risk Management",
+    "50": "Water in Agriculture",
+    "51": "Livestock and Sustainability",
+    "52": "Global Value Chains",
+    "53": "Competition Policy",
+    "54": "Pollution",
+    "55": "Urban Development",
+    "56": "Gender",
+    "57": "Safety Nets and Cash Transfers",
+    "58": "Forests and Landscapes",
+    "59": "Water Resources Management",
+    "60": "Extractive Industries",
+    "61": "Social Sustainability and Inclusion",
+    "62": "Energy",
+    "63": "Girls' Education",
+    "64": "Environment",
+    "65": "Marine Plastic Pollution",
+    "66": "Education and Technology",
+    "67": "Financial Integrity",
+    "68": "Oceans, Fisheries, and Coastal Economies",
+    "69": "Sanitation",
+    "70": "Land",
+    "71": "Higher Education",
+    "72": "Teachers",
+    "73": "Investment Climate",
+    "74": "Debt",
+    "75": "Climate Change",
+    "76": "Trade",
+    "77": "Skills Development",
+    "78": "Agribusiness and Value Chains",
+    "79": "Climate-Smart Agriculture",
+    "80": "Pensions",
+    "81": "Infectious diseases and Vaccines",
+    "82": "Payment Systems",
+    "83": "Community-Driven Development",
+    "84": "Water",
+    "85": "Governance",
+    "86": "Macroeconomics"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Agribusiness and Value Chains": 32,
-    "Agriculture and Food": 56,
-    "Biodiversity": 79,
-    "COVID-19 Hub": 60,
-    "Climate Change": 41,
-    "Climate-Smart Agriculture": 76,
-    "Community-Driven Development": 68,
-    "Competition Policy": 7,
-    "Competitiveness": 82,
-    "Credit Infrastructure": 69,
-    "Debt": 45,
-    "Debt Relief": 16,
-    "Digital Development": 75,
-    "Disability Inclusion": 33,
-    "Disaster Risk Management": 23,
-    "Early Childhood Development": 78,
-    "Education": 19,
-    "Education and Technology": 15,
-    "Energy": 14,
-    "Environment": 3,
-    "Extractive Industries": 54,
-    "Financial Inclusion": 24,
-    "Financial Integrity": 26,
-    "Financial Sector": 17,
-    "Financial Stability": 72,
-    "Food Security Update": 9,
-    "Food System Jobs": 77,
-    "Forests and Landscapes": 29,
-    "Fragility, Conflict, and Violence": 81,
-    "Gas Flaring Reduction": 59,
-    "Gender": 51,
-    "Girls' Education": 61,
-    "Global Financing Facility for Women, Children and Adolescents": 35,
-    "Global Value Chains": 55,
-    "Governance": 6,
-    "Health": 27,
-    "Higher Education": 40,
-    "Inclusive Cities": 53,
-    "Indigenous Peoples": 74,
-    "Inequality and Shared Prosperity": 11,
-    "Infectious diseases and Vaccines": 12,
-    "Infrastructure": 65,
-    "Innovation and Entrepreneurship": 48,
-    "Investment Climate": 2,
-    "Jobs & Development": 30,
-    "Jobs and Development": 13,
-    "Labor Markets": 5,
-    "Land": 38,
-    "Livestock and Sustainability": 22,
-    "Long-Term Finance": 8,
-    "Macroeconomics": 62,
-    "Marine Plastic Pollution": 34,
-    "Measuring Poverty": 18,
-    "Migration": 44,
-    "Mining Investment and Governance Review": 47,
-    "Natural Capital": 49,
-    "Nutrition": 84,
-    "Oceans, Fisheries, and Coastal Economies": 66,
-    "One Health": 73,
-    "Pandemic Preparedness and COVID-19": 64,
-    "Payment Systems": 31,
-    "Pensions": 71,
-    "Pollution": 70,
-    "Poverty": 80,
-    "Procurement for Development": 21,
-    "Regional Integration": 28,
-    "Safety Nets and Cash Transfers": 52,
-    "Sanitation": 58,
-    "Sexual Orientation and Gender Identity": 1,
-    "Skills Development": 36,
-    "Small and Medium Enterprises Finance": 42,
-    "Social Inclusion": 83,
-    "Social Protection": 10,
-    "Social Sustainability and Inclusion": 67,
-    "Sustainable Communities": 39,
-    "Sustainable Infrastructure Finance": 37,
-    "Taxes and Government Revenue": 85,
-    "Teachers": 43,
-    "Trade": 63,
-    "Trade Facilitation and Logistics": 25,
-    "Transport": 50,
-    "Universal Health Coverage": 57,
-    "Urban Development": 46,
-    "Water": 20,
-    "Water Resources Management": 4,
-    "Water Supply": 0,
-    "Water in Agriculture": 86
+    "Agribusiness and Value Chains": 78,
+    "Agriculture and Food": 18,
+    "Biodiversity": 22,
+    "COVID-19 Hub": 34,
+    "Climate Change": 75,
+    "Climate-Smart Agriculture": 79,
+    "Community-Driven Development": 83,
+    "Competition Policy": 53,
+    "Competitiveness": 1,
+    "Credit Infrastructure": 24,
+    "Debt": 74,
+    "Debt Relief": 48,
+    "Digital Development": 37,
+    "Disability Inclusion": 17,
+    "Disaster Risk Management": 49,
+    "Early Childhood Development": 26,
+    "Education": 3,
+    "Education and Technology": 66,
+    "Energy": 62,
+    "Environment": 64,
+    "Extractive Industries": 60,
+    "Financial Inclusion": 46,
+    "Financial Integrity": 67,
+    "Financial Sector": 31,
+    "Financial Stability": 36,
+    "Food Security Update": 16,
+    "Food System Jobs": 27,
+    "Forests and Landscapes": 58,
+    "Fragility, Conflict, and Violence": 47,
+    "Gas Flaring Reduction": 39,
+    "Gender": 56,
+    "Girls' Education": 63,
+    "Global Financing Facility for Women, Children and Adolescents": 0,
+    "Global Value Chains": 52,
+    "Governance": 85,
+    "Health": 43,
+    "Higher Education": 71,
+    "Inclusive Cities": 8,
+    "Indigenous Peoples": 29,
+    "Inequality and Shared Prosperity": 33,
+    "Infectious diseases and Vaccines": 81,
+    "Infrastructure": 42,
+    "Innovation and Entrepreneurship": 7,
+    "Investment Climate": 73,
+    "Jobs & Development": 9,
+    "Jobs and Development": 6,
+    "Labor Markets": 14,
+    "Land": 70,
+    "Livestock and Sustainability": 51,
+    "Long-Term Finance": 38,
+    "Macroeconomics": 86,
+    "Marine Plastic Pollution": 65,
+    "Measuring Poverty": 15,
+    "Migration": 28,
+    "Mining Investment and Governance Review": 40,
+    "Natural Capital": 4,
+    "Nutrition": 45,
+    "Oceans, Fisheries, and Coastal Economies": 68,
+    "One Health": 21,
+    "Pandemic Preparedness and COVID-19": 12,
+    "Payment Systems": 82,
+    "Pensions": 80,
+    "Pollution": 54,
+    "Poverty": 35,
+    "Procurement for Development": 32,
+    "Regional Integration": 5,
+    "Safety Nets and Cash Transfers": 57,
+    "Sanitation": 69,
+    "Sexual Orientation and Gender Identity": 44,
+    "Skills Development": 77,
+    "Small and Medium Enterprises Finance": 41,
+    "Social Inclusion": 23,
+    "Social Protection": 20,
+    "Social Sustainability and Inclusion": 61,
+    "Sustainable Communities": 2,
+    "Sustainable Infrastructure Finance": 11,
+    "Taxes and Government Revenue": 13,
+    "Teachers": 72,
+    "Trade": 76,
+    "Trade Facilitation and Logistics": 19,
+    "Transport": 10,
+    "Universal Health Coverage": 30,
+    "Urban Development": 55,
+    "Water": 84,
+    "Water Resources Management": 59,
+    "Water Supply": 25,
+    "Water in Agriculture": 50
   },
   "layer_norm_eps": 1e-07,
   "max_position_embeddings": 512,

last-checkpoint/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:47d9450ee32445570ac0f99bd61548d0db965a4e6edd93b15e3b6aaa0d756f1e
+oid sha256:171435965d16e8467d93cb72739550b78deb4e15f86639ad2ddc07a18249ce1e
 size 567860028

last-checkpoint/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9f74756958baa31a23d6968c507d7850e8689def8038cd22b1d5acff64dbc0db
+oid sha256:ac4231528fab4650ae11af5c79ae4d0c4ccc8c5d483a1dde1456cc539f94f31b
 size 1135783354

last-checkpoint/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:712d1e9e9623896c394d82d0be1b750fd445034b8cb9f0073258005a6df45273
+oid sha256:34649568145f465443beb487f2295ab3dad9e5f49f758646dd823029413e18fe
 size 14244

last-checkpoint/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:71d98b2235723fe888fb281dc5e6454bb5d9e489547cbc29dc5479c4d9a3698e
+oid sha256:8e136870ce715682b288c29fb4d05a0aec61f1a74b2cc393d7fb9e66e4965261
 size 1064

last-checkpoint/trainer_state.json

@@ -1,306 +1,55 @@
 {
-  "best_metric": 0.03788253664970398,
-  "best_model_checkpoint": "doc-topic-model_eval-00_train-02/checkpoint-11000",
-  "epoch": 5.424063116370808,
+  "best_metric": 0.0863451287150383,
+  "best_model_checkpoint": "doc-topic-model_eval-00_train-02/checkpoint-1000",
+  "epoch": 0.4930966469428008,
   "eval_steps": 1000,
-  "global_step": 11000,
+  "global_step": 1000,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
       "epoch": 0.2465483234714004,
-      "grad_norm": 0.3801194727420807,
-      "learning_rate": 1.95069033530572e-05,
-      "loss": 0.1664,
+      "grad_norm": 0.3767470121383667,
+      "learning_rate": 1.9950690335305722e-05,
+      "loss": 0.1663,
       "step": 500
     },
     {
       "epoch": 0.4930966469428008,
-      "grad_norm": 0.38533496856689453,
-      "learning_rate": 1.90138067061144e-05,
-      "loss": 0.0941,
+      "grad_norm": 0.4007192850112915,
+      "learning_rate": 1.9901380670611442e-05,
+      "loss": 0.0932,
       "step": 1000
     },
     {
       "epoch": 0.4930966469428008,
       "eval_accuracy": 0.9814660487265615,
       "eval_f1": 0.0,
-      "eval_loss": 0.08822464942932129,
+      "eval_loss": 0.0863451287150383,
       "eval_precision": 0.0,
       "eval_recall": 0.0,
-      "eval_runtime": 25.7552,
-      "eval_samples_per_second": 314.888,
-      "eval_steps_per_second": 19.685,
+      "eval_runtime": 11.8139,
+      "eval_samples_per_second": 686.478,
+      "eval_steps_per_second": 2.709,
       "step": 1000
-    },
-    {
-      "epoch": 0.7396449704142012,
-      "grad_norm": 0.38025280833244324,
-      "learning_rate": 1.85207100591716e-05,
-      "loss": 0.0858,
-      "step": 1500
-    },
-    {
-      "epoch": 0.9861932938856016,
-      "grad_norm": 0.37026312947273254,
-      "learning_rate": 1.80276134122288e-05,
-      "loss": 0.0776,
-      "step": 2000
-    },
-    {
-      "epoch": 0.9861932938856016,
-      "eval_accuracy": 0.9814660487265615,
-      "eval_f1": 0.0,
-      "eval_loss": 0.06934941560029984,
-      "eval_precision": 0.0,
-      "eval_recall": 0.0,
-      "eval_runtime": 26.0587,
-      "eval_samples_per_second": 311.22,
-      "eval_steps_per_second": 19.456,
-      "step": 2000
-    },
-    {
-      "epoch": 1.232741617357002,
-      "grad_norm": 0.42509472370147705,
-      "learning_rate": 1.7534516765285997e-05,
-      "loss": 0.0688,
-      "step": 2500
-    },
-    {
-      "epoch": 1.4792899408284024,
-      "grad_norm": 0.3655988276004791,
-      "learning_rate": 1.70414201183432e-05,
-      "loss": 0.0628,
-      "step": 3000
-    },
-    {
-      "epoch": 1.4792899408284024,
-      "eval_accuracy": 0.9822214663321853,
-      "eval_f1": 0.1038719817116731,
-      "eval_loss": 0.05777422711253166,
-      "eval_precision": 0.7893593919652552,
-      "eval_recall": 0.05559379062476103,
-      "eval_runtime": 26.8512,
-      "eval_samples_per_second": 302.035,
-      "eval_steps_per_second": 18.882,
-      "step": 3000
-    },
-    {
-      "epoch": 1.725838264299803,
-      "grad_norm": 0.37198927998542786,
-      "learning_rate": 1.6548323471400396e-05,
-      "loss": 0.0581,
-      "step": 3500
-    },
-    {
-      "epoch": 1.972386587771203,
-      "grad_norm": 0.34727558493614197,
-      "learning_rate": 1.6055226824457594e-05,
-      "loss": 0.0549,
-      "step": 4000
-    },
-    {
-      "epoch": 1.972386587771203,
-      "eval_accuracy": 0.9840894595858668,
-      "eval_f1": 0.300647894343384,
-      "eval_loss": 0.050482869148254395,
-      "eval_precision": 0.8110924369747899,
-      "eval_recall": 0.18452244398562362,
-      "eval_runtime": 26.4214,
-      "eval_samples_per_second": 306.948,
-      "eval_steps_per_second": 19.189,
-      "step": 4000
-    },
-    {
-      "epoch": 2.2189349112426036,
-      "grad_norm": 0.39387047290802,
-      "learning_rate": 1.5562130177514792e-05,
-      "loss": 0.0493,
-      "step": 4500
-    },
-    {
-      "epoch": 2.465483234714004,
-      "grad_norm": 0.32869574427604675,
-      "learning_rate": 1.5069033530571993e-05,
-      "loss": 0.0486,
-      "step": 5000
-    },
-    {
-      "epoch": 2.465483234714004,
-      "eval_accuracy": 0.9852473886361381,
-      "eval_f1": 0.4197558392329562,
-      "eval_loss": 0.04679542034864426,
-      "eval_precision": 0.7743726861373921,
-      "eval_recall": 0.2879100711172287,
-      "eval_runtime": 27.7172,
-      "eval_samples_per_second": 292.598,
-      "eval_steps_per_second": 18.292,
-      "step": 5000
-    },
-    {
-      "epoch": 2.712031558185404,
-      "grad_norm": 0.44469282031059265,
-      "learning_rate": 1.4575936883629191e-05,
-      "loss": 0.047,
-      "step": 5500
-    },
-    {
-      "epoch": 2.9585798816568047,
-      "grad_norm": 0.34966567158699036,
-      "learning_rate": 1.4082840236686392e-05,
-      "loss": 0.0456,
-      "step": 6000
-    },
-    {
-      "epoch": 2.9585798816568047,
-      "eval_accuracy": 0.9862536672477571,
-      "eval_f1": 0.47643724696356277,
-      "eval_loss": 0.04318871349096298,
-      "eval_precision": 0.8100220264317181,
-      "eval_recall": 0.33746272080752465,
-      "eval_runtime": 27.2164,
-      "eval_samples_per_second": 297.982,
-      "eval_steps_per_second": 18.628,
-      "step": 6000
-    },
-    {
-      "epoch": 3.2051282051282053,
-      "grad_norm": 0.3240692615509033,
-      "learning_rate": 1.3589743589743592e-05,
-      "loss": 0.0411,
-      "step": 6500
-    },
-    {
-      "epoch": 3.4516765285996054,
-      "grad_norm": 0.3621165454387665,
-      "learning_rate": 1.309664694280079e-05,
-      "loss": 0.0409,
-      "step": 7000
-    },
-    {
-      "epoch": 3.4516765285996054,
-      "eval_accuracy": 0.9866760208058732,
-      "eval_f1": 0.5272791270679338,
-      "eval_loss": 0.041621919721364975,
-      "eval_precision": 0.7698972099853157,
-      "eval_recall": 0.4009329356886136,
-      "eval_runtime": 26.6306,
-      "eval_samples_per_second": 304.537,
-      "eval_steps_per_second": 19.038,
-      "step": 7000
-    },
-    {
-      "epoch": 3.698224852071006,
-      "grad_norm": 0.4138104319572449,
-      "learning_rate": 1.2603550295857989e-05,
-      "loss": 0.0403,
-      "step": 7500
-    },
-    {
-      "epoch": 3.9447731755424065,
-      "grad_norm": 0.42052382230758667,
-      "learning_rate": 1.2110453648915189e-05,
-      "loss": 0.0388,
-      "step": 8000
-    },
-    {
-      "epoch": 3.9447731755424065,
-      "eval_accuracy": 0.9869098742860383,
-      "eval_f1": 0.5476097178683386,
-      "eval_loss": 0.04007947817444801,
-      "eval_precision": 0.7616841531543808,
-      "eval_recall": 0.4274680737172134,
-      "eval_runtime": 26.5409,
-      "eval_samples_per_second": 305.567,
-      "eval_steps_per_second": 19.103,
-      "step": 8000
-    },
-    {
-      "epoch": 4.191321499013807,
-      "grad_norm": 0.48471179604530334,
-      "learning_rate": 1.1617357001972386e-05,
-      "loss": 0.0376,
-      "step": 8500
-    },
-    {
-      "epoch": 4.437869822485207,
-      "grad_norm": 0.3329317271709442,
-      "learning_rate": 1.1124260355029586e-05,
-      "loss": 0.0356,
-      "step": 9000
-    },
-    {
-      "epoch": 4.437869822485207,
-      "eval_accuracy": 0.9870360134359454,
-      "eval_f1": 0.5559924275520606,
-      "eval_loss": 0.039172105491161346,
-      "eval_precision": 0.761164274322169,
-      "eval_recall": 0.43794448267951364,
-      "eval_runtime": 26.6005,
-      "eval_samples_per_second": 304.881,
-      "eval_steps_per_second": 19.06,
-      "step": 9000
-    },
-    {
-      "epoch": 4.684418145956608,
-      "grad_norm": 0.43250563740730286,
-      "learning_rate": 1.0631163708086787e-05,
-      "loss": 0.0352,
-      "step": 9500
-    },
-    {
-      "epoch": 4.930966469428008,
-      "grad_norm": 0.49166053533554077,
-      "learning_rate": 1.0138067061143987e-05,
-      "loss": 0.035,
-      "step": 10000
-    },
-    {
-      "epoch": 4.930966469428008,
-      "eval_accuracy": 0.9873464007823461,
-      "eval_f1": 0.5759475634083785,
-      "eval_loss": 0.03805544227361679,
-      "eval_precision": 0.7600601729973674,
-      "eval_recall": 0.46363844918559305,
-      "eval_runtime": 26.4106,
-      "eval_samples_per_second": 307.074,
-      "eval_steps_per_second": 19.197,
-      "step": 10000
-    },
-    {
-      "epoch": 5.177514792899408,
-      "grad_norm": 0.562140703201294,
-      "learning_rate": 9.644970414201184e-06,
-      "loss": 0.033,
-      "step": 10500
-    },
-    {
-      "epoch": 5.424063116370808,
-      "grad_norm": 0.41164836287498474,
-      "learning_rate": 9.151873767258384e-06,
-      "loss": 0.0316,
-      "step": 11000
-    },
-    {
-      "epoch": 5.424063116370808,
-      "eval_accuracy": 0.9873619910143572,
-      "eval_f1": 0.5883764944836819,
-      "eval_loss": 0.03788253664970398,
-      "eval_precision": 0.742254833449802,
-      "eval_recall": 0.4873441920929877,
-      "eval_runtime": 26.6635,
-      "eval_samples_per_second": 304.161,
-      "eval_steps_per_second": 19.015,
-      "step": 11000
     }
   ],
   "logging_steps": 500,
-  "max_steps": 20280,
+  "max_steps": 202800,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 10,
+  "num_train_epochs": 100,
   "save_steps": 1000,
   "stateful_callbacks": {
+    "EarlyStoppingCallback": {
+      "args": {
+        "early_stopping_patience": 5,
+        "early_stopping_threshold": 0.0
+      },
+      "attributes": {
+        "early_stopping_patience_counter": 0
+      }
+    },
     "TrainerControl": {
       "args": {
         "should_epoch_stop": false,
@@ -312,7 +61,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 248295812264388.0,
+  "total_flos": 22565797728696.0,
   "train_batch_size": 4,
   "trial_name": null,
   "trial_params": null

last-checkpoint/training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4b15ea728e9972258bd6817d47dab7216621d086d49df9d5c90ce4cef3be12a5
+oid sha256:52f3fd8049a89729a3c1cf7bfe4d3a85504bb9c099d007b31c6e5cbeb3e690e4
 size 5240