Training in progress, epoch 1

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:244c32ef5c6373ef12161581b12f3c30ace5bf6c6e7a386d865a53ade913778d
+oid sha256:b153083e3247a53ddfa45f8ace862dd036680a0af16004e032aae994694c3ccb
 size 438007864

run-3/checkpoint-2000/config.json

@@ -0,0 +1,65 @@
+{
+  "architectures": [
+    "BertForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3",
+    "4": "LABEL_4",
+    "5": "LABEL_5",
+    "6": "LABEL_6",
+    "7": "LABEL_7",
+    "8": "LABEL_8",
+    "9": "LABEL_9",
+    "10": "LABEL_10",
+    "11": "LABEL_11",
+    "12": "LABEL_12",
+    "13": "LABEL_13",
+    "14": "LABEL_14",
+    "15": "LABEL_15",
+    "16": "LABEL_16",
+    "17": "LABEL_17"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_10": 10,
+    "LABEL_11": 11,
+    "LABEL_12": 12,
+    "LABEL_13": 13,
+    "LABEL_14": 14,
+    "LABEL_15": 15,
+    "LABEL_16": 16,
+    "LABEL_17": 17,
+    "LABEL_2": 2,
+    "LABEL_3": 3,
+    "LABEL_4": 4,
+    "LABEL_5": 5,
+    "LABEL_6": 6,
+    "LABEL_7": 7,
+    "LABEL_8": 8,
+    "LABEL_9": 9
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "torch_dtype": "float32",
+  "transformers_version": "4.55.4",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 30522
+}

run-3/checkpoint-2000/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2940dfa4cafe155b3a07f3277464d1ed6c5c6abfc703787e7f1c6c2a80eccc34
+size 438007864

run-3/checkpoint-2000/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:015e603ee940575cef6daa23bdba26d2c24af9b32eee7b4b9c381357bfa9ede8
+size 876140043

run-3/checkpoint-2000/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dbbd0339d07bd579254b6321f2d4bfb420e0466d9c9ac3704bf31ec625375302
+size 14645

run-3/checkpoint-2000/scaler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6bd0201d3a3ae70736750a15db74e6d2f2b83f42bb2fa3ba3d4422088257b7d
+size 1383

run-3/checkpoint-2000/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbe852de2267093c96e4d1096f6ee2cc697881ff4730f6936b68e02375890ab2
+size 1465

run-3/checkpoint-2000/trainer_state.json

@@ -0,0 +1,278 @@
+{
+  "best_global_step": 1500,
+  "best_metric": 0.7817611579593701,
+  "best_model_checkpoint": "./results/run-3/checkpoint-1500",
+  "epoch": 4.0,
+  "eval_steps": 500,
+  "global_step": 2000,
+  "is_hyper_param_search": true,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.3840266466140747,
+      "learning_rate": 5.623773985869002e-05,
+      "loss": 0.4015,
+      "step": 100
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 36.44963455200195,
+      "learning_rate": 6.805455705058522e-05,
+      "loss": 0.0,
+      "step": 200
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.6683640480041504,
+      "learning_rate": 6.718217457333122e-05,
+      "loss": 0.0,
+      "step": 300
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.7861915230751038,
+      "learning_rate": 6.630979209607721e-05,
+      "loss": 0.1644,
+      "step": 400
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 2.053349256515503,
+      "learning_rate": 6.54374096188232e-05,
+      "loss": 0.0025,
+      "step": 500
+    },
+    {
+      "epoch": 1.0,
+      "eval_economic_inequality_accuracy": 0.7770780856423174,
+      "eval_economic_inequality_f1": 0.6795991195482776,
+      "eval_economic_policy_benefits_accuracy": 0.8215488215488216,
+      "eval_economic_policy_benefits_f1": 0.7410643713786043,
+      "eval_ethnic_boundaries_accuracy": 0.946078431372549,
+      "eval_ethnic_boundaries_f1": 0.9198646713093299,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.545923632610939,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.6805555555555556,
+      "eval_overall_accuracy": 0.8064020197505432,
+      "eval_overall_f1": 0.7222001215424422,
+      "eval_religion_ethnic_identity_accuracy": 0.839458413926499,
+      "eval_religion_ethnic_identity_f1": 0.7661933788519465,
+      "eval_runtime": 4.4769,
+      "eval_samples_per_second": 893.468,
+      "eval_steps_per_second": 55.842,
+      "step": 500
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 4.278098106384277,
+      "learning_rate": 6.456502714156919e-05,
+      "loss": 0.003,
+      "step": 600
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 2.1036500930786133,
+      "learning_rate": 6.369264466431519e-05,
+      "loss": 0.0604,
+      "step": 700
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 1.2531561851501465,
+      "learning_rate": 6.282026218706118e-05,
+      "loss": 0.0,
+      "step": 800
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 1.4897408485412598,
+      "learning_rate": 6.194787970980717e-05,
+      "loss": 0.0114,
+      "step": 900
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 1.9127349853515625,
+      "learning_rate": 6.107549723255315e-05,
+      "loss": 25991.0925,
+      "step": 1000
+    },
+    {
+      "epoch": 2.0,
+      "eval_economic_inequality_accuracy": 0.8224181360201511,
+      "eval_economic_inequality_f1": 0.7973444391161673,
+      "eval_economic_policy_benefits_accuracy": 0.8215488215488216,
+      "eval_economic_policy_benefits_f1": 0.7410643713786043,
+      "eval_ethnic_boundaries_accuracy": 0.9330065359477124,
+      "eval_ethnic_boundaries_f1": 0.9306378882302275,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.545923632610939,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.6805555555555556,
+      "eval_overall_accuracy": 0.8117800455760427,
+      "eval_overall_f1": 0.7478535187845178,
+      "eval_religion_ethnic_identity_accuracy": 0.839458413926499,
+      "eval_religion_ethnic_identity_f1": 0.7915952258156131,
+      "eval_runtime": 4.4757,
+      "eval_samples_per_second": 893.725,
+      "eval_steps_per_second": 55.858,
+      "step": 1000
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 0.9552398324012756,
+      "learning_rate": 6.020311475529915e-05,
+      "loss": 0.0,
+      "step": 1100
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 1.114319920539856,
+      "learning_rate": 5.9330732278045136e-05,
+      "loss": 0.0009,
+      "step": 1200
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 0.16842737793922424,
+      "learning_rate": 5.845834980079113e-05,
+      "loss": 0.0,
+      "step": 1300
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 1.0355138778686523,
+      "learning_rate": 5.758596732353712e-05,
+      "loss": 0.0638,
+      "step": 1400
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 0.5731643438339233,
+      "learning_rate": 5.6713584846283115e-05,
+      "loss": 0.0,
+      "step": 1500
+    },
+    {
+      "epoch": 3.0,
+      "eval_economic_inequality_accuracy": 0.8224181360201511,
+      "eval_economic_inequality_f1": 0.7938100421450593,
+      "eval_economic_policy_benefits_accuracy": 0.8383838383838383,
+      "eval_economic_policy_benefits_f1": 0.8108014554114583,
+      "eval_ethnic_boundaries_accuracy": 0.9395424836601307,
+      "eval_ethnic_boundaries_f1": 0.9340440546398507,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.6109373423468875,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.7407407407407408,
+      "eval_overall_accuracy": 0.8150304610083521,
+      "eval_overall_f1": 0.7817611579593701,
+      "eval_religion_ethnic_identity_accuracy": 0.8355899419729207,
+      "eval_religion_ethnic_identity_f1": 0.8002333124722241,
+      "eval_runtime": 4.4858,
+      "eval_samples_per_second": 891.71,
+      "eval_steps_per_second": 55.732,
+      "step": 1500
+    },
+    {
+      "epoch": 3.2,
+      "grad_norm": 0.58502596616745,
+      "learning_rate": 5.58412023690291e-05,
+      "loss": 0.0013,
+      "step": 1600
+    },
+    {
+      "epoch": 3.4,
+      "grad_norm": 12.144497871398926,
+      "learning_rate": 5.49688198917751e-05,
+      "loss": 0.0,
+      "step": 1700
+    },
+    {
+      "epoch": 3.6,
+      "grad_norm": 0.9917765259742737,
+      "learning_rate": 5.4096437414521086e-05,
+      "loss": 0.1593,
+      "step": 1800
+    },
+    {
+      "epoch": 3.8,
+      "grad_norm": 0.2664477825164795,
+      "learning_rate": 5.3224054937267074e-05,
+      "loss": 0.0082,
+      "step": 1900
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 0.33486196398735046,
+      "learning_rate": 5.235167246001307e-05,
+      "loss": 0.0,
+      "step": 2000
+    },
+    {
+      "epoch": 4.0,
+      "eval_economic_inequality_accuracy": 0.783375314861461,
+      "eval_economic_inequality_f1": 0.7832095348608167,
+      "eval_economic_policy_benefits_accuracy": 0.7845117845117845,
+      "eval_economic_policy_benefits_f1": 0.7745310880322321,
+      "eval_ethnic_boundaries_accuracy": 0.9101307189542484,
+      "eval_ethnic_boundaries_f1": 0.9195608338612992,
+      "eval_language_policy_accuracy": 0.6176470588235294,
+      "eval_language_policy_f1": 0.5673853251707923,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.7407407407407408,
+      "eval_overall_accuracy": 0.780003176208314,
+      "eval_overall_f1": 0.7667330042293844,
+      "eval_religion_ethnic_identity_accuracy": 0.8065764023210832,
+      "eval_religion_ethnic_identity_f1": 0.8149705027104255,
+      "eval_runtime": 4.4762,
+      "eval_samples_per_second": 893.625,
+      "eval_steps_per_second": 55.852,
+      "step": 2000
+    }
+  ],
+  "logging_steps": 100,
+  "max_steps": 8000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 16,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "EarlyStoppingCallback": {
+      "args": {
+        "early_stopping_patience": 2,
+        "early_stopping_threshold": 0.0
+      },
+      "attributes": {
+        "early_stopping_patience_counter": 1
+      }
+    },
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.6841526607872e+16,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": {
+    "gradient_accumulation_steps": 4,
+    "learning_rate": 6.873501538284335e-05,
+    "num_train_epochs": 16,
+    "per_device_train_batch_size": 8,
+    "warmup_steps": 121,
+    "weight_decay": 0.08748775333302956
+  }
+}

run-3/checkpoint-2000/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f25b0890fc1a55f215dd25d4ee6812fa588a4d653eedd59fd60c8d3a14cbe61a
+size 5777

run-3/checkpoint-2500/config.json

@@ -0,0 +1,65 @@
+{
+  "architectures": [
+    "BertForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3",
+    "4": "LABEL_4",
+    "5": "LABEL_5",
+    "6": "LABEL_6",
+    "7": "LABEL_7",
+    "8": "LABEL_8",
+    "9": "LABEL_9",
+    "10": "LABEL_10",
+    "11": "LABEL_11",
+    "12": "LABEL_12",
+    "13": "LABEL_13",
+    "14": "LABEL_14",
+    "15": "LABEL_15",
+    "16": "LABEL_16",
+    "17": "LABEL_17"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_10": 10,
+    "LABEL_11": 11,
+    "LABEL_12": 12,
+    "LABEL_13": 13,
+    "LABEL_14": 14,
+    "LABEL_15": 15,
+    "LABEL_16": 16,
+    "LABEL_17": 17,
+    "LABEL_2": 2,
+    "LABEL_3": 3,
+    "LABEL_4": 4,
+    "LABEL_5": 5,
+    "LABEL_6": 6,
+    "LABEL_7": 7,
+    "LABEL_8": 8,
+    "LABEL_9": 9
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "torch_dtype": "float32",
+  "transformers_version": "4.55.4",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 30522
+}

run-3/checkpoint-2500/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ef89747ed4c4c141693b9a8883889b43c13d17f1d95fa9a4604cc27bfb4fb57
+size 438007864

run-3/checkpoint-2500/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bbad1a481c3e3a739b7e21f7b7649c119ddce3f6e3bda33f1b1db570a2b770e1
+size 876140043

run-3/checkpoint-2500/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14c39e64e59e8286453bf0b17c32f65ff9b3c161db8386dcfa997c86bd201138
+size 14645

run-3/checkpoint-2500/scaler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bdcdf4c7d42b8a27a5154f492b6c963d35ee2201f61c4484dda1057a0bfc22a5
+size 1383

run-3/checkpoint-2500/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3c5c3ccdcf1bbdfee4f7e711d0ce76d0fd57086fa53bb522a13c17067824d970
+size 1465

run-3/checkpoint-2500/trainer_state.json

@@ -0,0 +1,335 @@
+{
+  "best_global_step": 1500,
+  "best_metric": 0.7817611579593701,
+  "best_model_checkpoint": "./results/run-3/checkpoint-1500",
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 2500,
+  "is_hyper_param_search": true,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.3840266466140747,
+      "learning_rate": 5.623773985869002e-05,
+      "loss": 0.4015,
+      "step": 100
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 36.44963455200195,
+      "learning_rate": 6.805455705058522e-05,
+      "loss": 0.0,
+      "step": 200
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.6683640480041504,
+      "learning_rate": 6.718217457333122e-05,
+      "loss": 0.0,
+      "step": 300
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.7861915230751038,
+      "learning_rate": 6.630979209607721e-05,
+      "loss": 0.1644,
+      "step": 400
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 2.053349256515503,
+      "learning_rate": 6.54374096188232e-05,
+      "loss": 0.0025,
+      "step": 500
+    },
+    {
+      "epoch": 1.0,
+      "eval_economic_inequality_accuracy": 0.7770780856423174,
+      "eval_economic_inequality_f1": 0.6795991195482776,
+      "eval_economic_policy_benefits_accuracy": 0.8215488215488216,
+      "eval_economic_policy_benefits_f1": 0.7410643713786043,
+      "eval_ethnic_boundaries_accuracy": 0.946078431372549,
+      "eval_ethnic_boundaries_f1": 0.9198646713093299,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.545923632610939,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.6805555555555556,
+      "eval_overall_accuracy": 0.8064020197505432,
+      "eval_overall_f1": 0.7222001215424422,
+      "eval_religion_ethnic_identity_accuracy": 0.839458413926499,
+      "eval_religion_ethnic_identity_f1": 0.7661933788519465,
+      "eval_runtime": 4.4769,
+      "eval_samples_per_second": 893.468,
+      "eval_steps_per_second": 55.842,
+      "step": 500
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 4.278098106384277,
+      "learning_rate": 6.456502714156919e-05,
+      "loss": 0.003,
+      "step": 600
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 2.1036500930786133,
+      "learning_rate": 6.369264466431519e-05,
+      "loss": 0.0604,
+      "step": 700
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 1.2531561851501465,
+      "learning_rate": 6.282026218706118e-05,
+      "loss": 0.0,
+      "step": 800
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 1.4897408485412598,
+      "learning_rate": 6.194787970980717e-05,
+      "loss": 0.0114,
+      "step": 900
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 1.9127349853515625,
+      "learning_rate": 6.107549723255315e-05,
+      "loss": 25991.0925,
+      "step": 1000
+    },
+    {
+      "epoch": 2.0,
+      "eval_economic_inequality_accuracy": 0.8224181360201511,
+      "eval_economic_inequality_f1": 0.7973444391161673,
+      "eval_economic_policy_benefits_accuracy": 0.8215488215488216,
+      "eval_economic_policy_benefits_f1": 0.7410643713786043,
+      "eval_ethnic_boundaries_accuracy": 0.9330065359477124,
+      "eval_ethnic_boundaries_f1": 0.9306378882302275,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.545923632610939,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.6805555555555556,
+      "eval_overall_accuracy": 0.8117800455760427,
+      "eval_overall_f1": 0.7478535187845178,
+      "eval_religion_ethnic_identity_accuracy": 0.839458413926499,
+      "eval_religion_ethnic_identity_f1": 0.7915952258156131,
+      "eval_runtime": 4.4757,
+      "eval_samples_per_second": 893.725,
+      "eval_steps_per_second": 55.858,
+      "step": 1000
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 0.9552398324012756,
+      "learning_rate": 6.020311475529915e-05,
+      "loss": 0.0,
+      "step": 1100
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 1.114319920539856,
+      "learning_rate": 5.9330732278045136e-05,
+      "loss": 0.0009,
+      "step": 1200
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 0.16842737793922424,
+      "learning_rate": 5.845834980079113e-05,
+      "loss": 0.0,
+      "step": 1300
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 1.0355138778686523,
+      "learning_rate": 5.758596732353712e-05,
+      "loss": 0.0638,
+      "step": 1400
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 0.5731643438339233,
+      "learning_rate": 5.6713584846283115e-05,
+      "loss": 0.0,
+      "step": 1500
+    },
+    {
+      "epoch": 3.0,
+      "eval_economic_inequality_accuracy": 0.8224181360201511,
+      "eval_economic_inequality_f1": 0.7938100421450593,
+      "eval_economic_policy_benefits_accuracy": 0.8383838383838383,
+      "eval_economic_policy_benefits_f1": 0.8108014554114583,
+      "eval_ethnic_boundaries_accuracy": 0.9395424836601307,
+      "eval_ethnic_boundaries_f1": 0.9340440546398507,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.6109373423468875,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.7407407407407408,
+      "eval_overall_accuracy": 0.8150304610083521,
+      "eval_overall_f1": 0.7817611579593701,
+      "eval_religion_ethnic_identity_accuracy": 0.8355899419729207,
+      "eval_religion_ethnic_identity_f1": 0.8002333124722241,
+      "eval_runtime": 4.4858,
+      "eval_samples_per_second": 891.71,
+      "eval_steps_per_second": 55.732,
+      "step": 1500
+    },
+    {
+      "epoch": 3.2,
+      "grad_norm": 0.58502596616745,
+      "learning_rate": 5.58412023690291e-05,
+      "loss": 0.0013,
+      "step": 1600
+    },
+    {
+      "epoch": 3.4,
+      "grad_norm": 12.144497871398926,
+      "learning_rate": 5.49688198917751e-05,
+      "loss": 0.0,
+      "step": 1700
+    },
+    {
+      "epoch": 3.6,
+      "grad_norm": 0.9917765259742737,
+      "learning_rate": 5.4096437414521086e-05,
+      "loss": 0.1593,
+      "step": 1800
+    },
+    {
+      "epoch": 3.8,
+      "grad_norm": 0.2664477825164795,
+      "learning_rate": 5.3224054937267074e-05,
+      "loss": 0.0082,
+      "step": 1900
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 0.33486196398735046,
+      "learning_rate": 5.235167246001307e-05,
+      "loss": 0.0,
+      "step": 2000
+    },
+    {
+      "epoch": 4.0,
+      "eval_economic_inequality_accuracy": 0.783375314861461,
+      "eval_economic_inequality_f1": 0.7832095348608167,
+      "eval_economic_policy_benefits_accuracy": 0.7845117845117845,
+      "eval_economic_policy_benefits_f1": 0.7745310880322321,
+      "eval_ethnic_boundaries_accuracy": 0.9101307189542484,
+      "eval_ethnic_boundaries_f1": 0.9195608338612992,
+      "eval_language_policy_accuracy": 0.6176470588235294,
+      "eval_language_policy_f1": 0.5673853251707923,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.7407407407407408,
+      "eval_overall_accuracy": 0.780003176208314,
+      "eval_overall_f1": 0.7667330042293844,
+      "eval_religion_ethnic_identity_accuracy": 0.8065764023210832,
+      "eval_religion_ethnic_identity_f1": 0.8149705027104255,
+      "eval_runtime": 4.4762,
+      "eval_samples_per_second": 893.625,
+      "eval_steps_per_second": 55.852,
+      "step": 2000
+    },
+    {
+      "epoch": 4.2,
+      "grad_norm": 0.5987537503242493,
+      "learning_rate": 5.147928998275906e-05,
+      "loss": 0.0,
+      "step": 2100
+    },
+    {
+      "epoch": 4.4,
+      "grad_norm": 0.334614634513855,
+      "learning_rate": 5.060690750550505e-05,
+      "loss": 0.0001,
+      "step": 2200
+    },
+    {
+      "epoch": 4.6,
+      "grad_norm": 1.4746909141540527,
+      "learning_rate": 4.973452502825104e-05,
+      "loss": 0.0277,
+      "step": 2300
+    },
+    {
+      "epoch": 4.8,
+      "grad_norm": 4.362077236175537,
+      "learning_rate": 4.8862142550997035e-05,
+      "loss": 0.0004,
+      "step": 2400
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 2.2466182708740234,
+      "learning_rate": 4.7989760073743024e-05,
+      "loss": 0.0,
+      "step": 2500
+    },
+    {
+      "epoch": 5.0,
+      "eval_economic_inequality_accuracy": 0.8047858942065491,
+      "eval_economic_inequality_f1": 0.7888912978895618,
+      "eval_economic_policy_benefits_accuracy": 0.8215488215488216,
+      "eval_economic_policy_benefits_f1": 0.7944704711827999,
+      "eval_ethnic_boundaries_accuracy": 0.9411764705882353,
+      "eval_ethnic_boundaries_f1": 0.937363686657517,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.5869281045751634,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.7407407407407408,
+      "eval_overall_accuracy": 0.8098806217177312,
+      "eval_overall_f1": 0.7790807091747901,
+      "eval_religion_ethnic_identity_accuracy": 0.8375241779497099,
+      "eval_religion_ethnic_identity_f1": 0.8260899540029579,
+      "eval_runtime": 4.4734,
+      "eval_samples_per_second": 894.178,
+      "eval_steps_per_second": 55.886,
+      "step": 2500
+    }
+  ],
+  "logging_steps": 100,
+  "max_steps": 8000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 16,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "EarlyStoppingCallback": {
+      "args": {
+        "early_stopping_patience": 2,
+        "early_stopping_threshold": 0.0
+      },
+      "attributes": {
+        "early_stopping_patience_counter": 2
+      }
+    },
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 2.105190825984e+16,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": {
+    "gradient_accumulation_steps": 4,
+    "learning_rate": 6.873501538284335e-05,
+    "num_train_epochs": 16,
+    "per_device_train_batch_size": 8,
+    "warmup_steps": 121,
+    "weight_decay": 0.08748775333302956
+  }
+}

run-3/checkpoint-2500/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f25b0890fc1a55f215dd25d4ee6812fa588a4d653eedd59fd60c8d3a14cbe61a
+size 5777

run-4/checkpoint-1000/config.json

@@ -0,0 +1,65 @@
+{
+  "architectures": [
+    "BertForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3",
+    "4": "LABEL_4",
+    "5": "LABEL_5",
+    "6": "LABEL_6",
+    "7": "LABEL_7",
+    "8": "LABEL_8",
+    "9": "LABEL_9",
+    "10": "LABEL_10",
+    "11": "LABEL_11",
+    "12": "LABEL_12",
+    "13": "LABEL_13",
+    "14": "LABEL_14",
+    "15": "LABEL_15",
+    "16": "LABEL_16",
+    "17": "LABEL_17"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_10": 10,
+    "LABEL_11": 11,
+    "LABEL_12": 12,
+    "LABEL_13": 13,
+    "LABEL_14": 14,
+    "LABEL_15": 15,
+    "LABEL_16": 16,
+    "LABEL_17": 17,
+    "LABEL_2": 2,
+    "LABEL_3": 3,
+    "LABEL_4": 4,
+    "LABEL_5": 5,
+    "LABEL_6": 6,
+    "LABEL_7": 7,
+    "LABEL_8": 8,
+    "LABEL_9": 9
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "torch_dtype": "float32",
+  "transformers_version": "4.55.4",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 30522
+}

run-4/checkpoint-1000/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b153083e3247a53ddfa45f8ace862dd036680a0af16004e032aae994694c3ccb
+size 438007864

run-4/checkpoint-1000/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7302841e07ab5eda5430dc0385819e43b7bf640bc93cb0738a885a8ec4a6a93
+size 876140043

run-4/checkpoint-1000/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:048b04c7b3a288839e6f9d3960cc6c47e42593576af476929e4d0adce308d3f2
+size 14645

run-4/checkpoint-1000/scaler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7f2f965d2db5576ee3b15eb73913964e83bfa88d286b89a8af0f82b0923633f
+size 1383

run-4/checkpoint-1000/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d63a7d843bba1326c5c800cfa088f77df4fab7ec4c535a9c7c5bf4694d26d305
+size 1465

run-4/checkpoint-1000/trainer_state.json

@@ -0,0 +1,142 @@
+{
+  "best_global_step": 1000,
+  "best_metric": 0.7331392486819855,
+  "best_model_checkpoint": "./results/run-4/checkpoint-1000",
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 1000,
+  "is_hyper_param_search": true,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.1,
+      "grad_norm": 3.002380609512329,
+      "learning_rate": 4.572943108695439e-06,
+      "loss": 0.296,
+      "step": 100
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.9367781281471252,
+      "learning_rate": 9.192077561923155e-06,
+      "loss": 1.0274,
+      "step": 200
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.7274576425552368,
+      "learning_rate": 1.3811212015150872e-05,
+      "loss": 0.0,
+      "step": 300
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 44.297061920166016,
+      "learning_rate": 1.8430346468378586e-05,
+      "loss": 0.0,
+      "step": 400
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 1.0151411294937134,
+      "learning_rate": 2.3049480921606306e-05,
+      "loss": 0.0,
+      "step": 500
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 1.208032488822937,
+      "learning_rate": 2.766861537483402e-05,
+      "loss": 0.0,
+      "step": 600
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 6.984821319580078,
+      "learning_rate": 3.2287749828061733e-05,
+      "loss": 0.0342,
+      "step": 700
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 1.5860588550567627,
+      "learning_rate": 3.690688428128945e-05,
+      "loss": 0.0488,
+      "step": 800
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.7841507196426392,
+      "learning_rate": 4.152601873451717e-05,
+      "loss": 0.0314,
+      "step": 900
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.9344207644462585,
+      "learning_rate": 4.414230747586468e-05,
+      "loss": 0.0356,
+      "step": 1000
+    },
+    {
+      "epoch": 1.0,
+      "eval_economic_inequality_accuracy": 0.7783375314861462,
+      "eval_economic_inequality_f1": 0.6829252350399464,
+      "eval_economic_policy_benefits_accuracy": 0.8215488215488216,
+      "eval_economic_policy_benefits_f1": 0.752833443442648,
+      "eval_ethnic_boundaries_accuracy": 0.9493464052287581,
+      "eval_ethnic_boundaries_f1": 0.9305506350938352,
+      "eval_language_policy_accuracy": 0.6764705882352942,
+      "eval_language_policy_f1": 0.555672268907563,
+      "eval_loss": NaN,
+      "eval_mother_tongue_education_accuracy": 0.7777777777777778,
+      "eval_mother_tongue_education_f1": 0.6805555555555556,
+      "eval_overall_accuracy": 0.8090908256773385,
+      "eval_overall_f1": 0.7331392486819855,
+      "eval_religion_ethnic_identity_accuracy": 0.851063829787234,
+      "eval_religion_ethnic_identity_f1": 0.7962983540523647,
+      "eval_runtime": 4.4716,
+      "eval_samples_per_second": 894.535,
+      "eval_steps_per_second": 55.908,
+      "step": 1000
+    }
+  ],
+  "logging_steps": 100,
+  "max_steps": 5000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "EarlyStoppingCallback": {
+      "args": {
+        "early_stopping_patience": 2,
+        "early_stopping_threshold": 0.0
+      },
+      "attributes": {
+        "early_stopping_patience_counter": 0
+      }
+    },
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 4210381651968000.0,
+  "train_batch_size": 16,
+  "trial_name": null,
+  "trial_params": {
+    "gradient_accumulation_steps": 1,
+    "learning_rate": 4.452845612911518e-05,
+    "num_train_epochs": 5,
+    "per_device_train_batch_size": 16,
+    "warmup_steps": 964,
+    "weight_decay": 0.08427803271887896
+  }
+}

run-4/checkpoint-1000/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a4ad559f091ae7757869f5224f652bb49e6e3989ffd03fbb0eefdb9252513c40
+size 5777

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f25b0890fc1a55f215dd25d4ee6812fa588a4d653eedd59fd60c8d3a14cbe61a
+oid sha256:a4ad559f091ae7757869f5224f652bb49e6e3989ffd03fbb0eefdb9252513c40
 size 5777