Upload folder using huggingface_hub

training_checkpoints/checkpoint-1929/config.json

@@ -0,0 +1,63 @@
+{
+  "_name_or_path": "openaccess-ai-collective/tiny-mistral",
+  "architectures": [
+    "MistralForSequenceClassification"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "dropout_p": 0.1,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 512,
+  "id2label": {
+    "0": "Issue",
+    "1": "Court Discourse",
+    "2": "Conclusion",
+    "3": "Precedent Analysis",
+    "4": "Section Analysis",
+    "5": "Argument by Petitioner",
+    "6": "Fact",
+    "7": "Argument by Respondent",
+    "8": "Ratio",
+    "9": "Appellant",
+    "10": "Respondent",
+    "11": "Argument by Appellant",
+    "12": "Petitioner",
+    "13": "Judge",
+    "14": "Argument by Defendant"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 14336,
+  "label2id": {
+    "Appellant": 9,
+    "Argument by Appellant": 11,
+    "Argument by Defendant": 14,
+    "Argument by Petitioner": 5,
+    "Argument by Respondent": 7,
+    "Conclusion": 2,
+    "Court Discourse": 1,
+    "Fact": 6,
+    "Issue": 0,
+    "Judge": 13,
+    "Petitioner": 12,
+    "Precedent Analysis": 3,
+    "Ratio": 8,
+    "Respondent": 10,
+    "Section Analysis": 4
+  },
+  "max_position_embeddings": 32768,
+  "model_type": "mistral",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 8,
+  "num_key_value_heads": 4,
+  "pad_token_id": 32000,
+  "problem_type": "single_label_classification",
+  "rms_norm_eps": 1e-05,
+  "rope_theta": 10000.0,
+  "sliding_window": 4096,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.39.3",
+  "use_cache": true,
+  "vocab_size": 32001
+}

training_checkpoints/checkpoint-1929/model.safetensors

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

training_checkpoints/checkpoint-1929/optimizer.pt

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

training_checkpoints/checkpoint-1929/rng_state.pth

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

training_checkpoints/checkpoint-1929/scheduler.pt

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

training_checkpoints/checkpoint-1929/trainer_state.json

@@ -0,0 +1,108 @@
+{
+  "best_metric": 0.6012751387344262,
+  "best_model_checkpoint": "tiny-mistral/checkpoint-1929",
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 1929,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.78,
+      "grad_norm": 13.069233894348145,
+      "learning_rate": 4.7413167444271646e-05,
+      "loss": 1.4479,
+      "step": 500
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.6498838109992254,
+      "eval_f1_macro": 0.45467871431664447,
+      "eval_f1_micro": 0.6498838109992254,
+      "eval_f1_weighted": 0.6214028764904677,
+      "eval_loss": 1.118202805519104,
+      "eval_macro_fpr": 0.039027492204575735,
+      "eval_macro_sensitivity": 0.4743680961365506,
+      "eval_macro_specificity": 0.9731278674075015,
+      "eval_precision": 0.6258370043663808,
+      "eval_precision_macro": 0.4712458606651466,
+      "eval_recall": 0.6498838109992254,
+      "eval_recall_macro": 0.4743680961365506,
+      "eval_runtime": 52.8829,
+      "eval_samples_per_second": 24.412,
+      "eval_steps_per_second": 3.063,
+      "eval_weighted_fpr": 0.03705525495982948,
+      "eval_weighted_sensitivity": 0.6498838109992254,
+      "eval_weighted_specificity": 0.9470342001132973,
+      "step": 643
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 42.20966720581055,
+      "learning_rate": 4.482115085536548e-05,
+      "loss": 0.8133,
+      "step": 1000
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.6986831913245546,
+      "eval_f1_macro": 0.5473679856051097,
+      "eval_f1_micro": 0.6986831913245546,
+      "eval_f1_weighted": 0.6969580552726153,
+      "eval_loss": 1.0854355096817017,
+      "eval_macro_fpr": 0.030488809177152586,
+      "eval_macro_sensitivity": 0.552765390116517,
+      "eval_macro_specificity": 0.9773020581125585,
+      "eval_precision": 0.7196764053623382,
+      "eval_precision_macro": 0.5876945028566362,
+      "eval_recall": 0.6986831913245546,
+      "eval_recall_macro": 0.552765390116517,
+      "eval_runtime": 93.8123,
+      "eval_samples_per_second": 13.762,
+      "eval_steps_per_second": 1.727,
+      "eval_weighted_fpr": 0.029883997848966736,
+      "eval_weighted_sensitivity": 0.6986831913245546,
+      "eval_weighted_specificity": 0.9608476803638234,
+      "step": 1286
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 26.4113712310791,
+      "learning_rate": 4.223950233281493e-05,
+      "loss": 0.5592,
+      "step": 1500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.6986831913245546,
+      "eval_f1_macro": 0.6012751387344262,
+      "eval_f1_micro": 0.6986831913245546,
+      "eval_f1_weighted": 0.6997681579317561,
+      "eval_loss": 1.6113793849945068,
+      "eval_macro_fpr": 0.030358926871943602,
+      "eval_macro_sensitivity": 0.5880627120151358,
+      "eval_macro_specificity": 0.9773067719065487,
+      "eval_precision": 0.7107021565435693,
+      "eval_precision_macro": 0.6368062038606529,
+      "eval_recall": 0.6986831913245546,
+      "eval_recall_macro": 0.5880627120151358,
+      "eval_runtime": 96.7771,
+      "eval_samples_per_second": 13.34,
+      "eval_steps_per_second": 1.674,
+      "eval_weighted_fpr": 0.029883997848966736,
+      "eval_weighted_sensitivity": 0.6986831913245546,
+      "eval_weighted_specificity": 0.9609183872736765,
+      "step": 1929
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 9645,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 15,
+  "save_steps": 500,
+  "total_flos": 8590566522617856.0,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": null
+}

training_checkpoints/checkpoint-1929/training_args.bin

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