Upload folder using huggingface_hub

training_checkpoints/checkpoint-1286/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-1286/model.safetensors

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

training_checkpoints/checkpoint-1286/optimizer.pt

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

training_checkpoints/checkpoint-1286/rng_state.pth

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

training_checkpoints/checkpoint-1286/scheduler.pt

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

training_checkpoints/checkpoint-1286/trainer_state.json

@@ -0,0 +1,79 @@
+{
+  "best_metric": 0.5473679856051097,
+  "best_model_checkpoint": "tiny-mistral/checkpoint-1286",
+  "epoch": 2.0,
+  "eval_steps": 500,
+  "global_step": 1286,
+  "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
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 9645,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 15,
+  "save_steps": 500,
+  "total_flos": 5727044348411904.0,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": null
+}

training_checkpoints/checkpoint-1286/training_args.bin

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