Training in progress, epoch 6

logs/events.out.tfevents.1719308784.852b1e905a9a.223.3

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f686028ea082d55a3a3697bf1e8edb9cf5d8e5562e25fda73a7f8c1ffa93311e
-size 7564
+oid sha256:844a0464575996ef9f317803cc9351f0115165050c50747adeb346cf4243d375
+size 8098

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:957a82fa99114649a04bc99d6cf73a08d75382fe915b3d4033b919814f1dc519
+oid sha256:050b2db8a7faf810b1b2e4395161325773cc1fe1b670a1f39b5cd6a63508710f
 size 17549312

run-3/checkpoint-3162/config.json

@@ -0,0 +1,34 @@
+{
+  "_name_or_path": "google/bert_uncased_L-2_H-128_A-2",
+  "architectures": [
+    "BertForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 128,
+  "id2label": {
+    "0": "negative",
+    "1": "positive"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 512,
+  "label2id": {
+    "negative": "0",
+    "positive": "1"
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 2,
+  "num_hidden_layers": 2,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "problem_type": "single_label_classification",
+  "torch_dtype": "float32",
+  "transformers_version": "4.41.2",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 30522
+}

run-3/checkpoint-3162/model.safetensors

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

run-3/checkpoint-3162/optimizer.pt

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

run-3/checkpoint-3162/rng_state.pth

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

run-3/checkpoint-3162/scheduler.pt

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

run-3/checkpoint-3162/special_tokens_map.json

@@ -0,0 +1,7 @@
+{
+  "cls_token": "[CLS]",
+  "mask_token": "[MASK]",
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "unk_token": "[UNK]"
+}

run-3/checkpoint-3162/tokenizer_config.json

@@ -0,0 +1,57 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "101": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "102": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "103": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "model_max_length": 512,
+  "never_split": null,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "unk_token": "[UNK]"
+}

run-3/checkpoint-3162/trainer_state.json

@@ -0,0 +1,134 @@
+{
+  "best_metric": 0.8325688073394495,
+  "best_model_checkpoint": "tiny-bert-sst2-distilled/run-3/checkpoint-3162",
+  "epoch": 6.0,
+  "eval_steps": 500,
+  "global_step": 3162,
+  "is_hyper_param_search": true,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "grad_norm": 31.228694915771484,
+      "learning_rate": 7.086160775881376e-05,
+      "loss": 2.5988,
+      "step": 527
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.8061926605504587,
+      "eval_loss": 1.924224615097046,
+      "eval_runtime": 2.5772,
+      "eval_samples_per_second": 338.351,
+      "eval_steps_per_second": 2.716,
+      "step": 527
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 51.49909591674805,
+      "learning_rate": 5.9055073680756574e-05,
+      "loss": 1.2849,
+      "step": 1054
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.8176605504587156,
+      "eval_loss": 1.7333961725234985,
+      "eval_runtime": 2.54,
+      "eval_samples_per_second": 343.313,
+      "eval_steps_per_second": 2.756,
+      "step": 1054
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 84.45722961425781,
+      "learning_rate": 4.72485396026994e-05,
+      "loss": 0.9617,
+      "step": 1581
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.8245412844036697,
+      "eval_loss": 1.7150405645370483,
+      "eval_runtime": 2.5578,
+      "eval_samples_per_second": 340.916,
+      "eval_steps_per_second": 2.737,
+      "step": 1581
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 30.98444938659668,
+      "learning_rate": 3.5442005524642224e-05,
+      "loss": 0.8154,
+      "step": 2108
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.8222477064220184,
+      "eval_loss": 1.6820627450942993,
+      "eval_runtime": 2.5749,
+      "eval_samples_per_second": 338.654,
+      "eval_steps_per_second": 2.719,
+      "step": 2108
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 18.188648223876953,
+      "learning_rate": 2.3635471446585048e-05,
+      "loss": 0.7233,
+      "step": 2635
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.8245412844036697,
+      "eval_loss": 1.7338383197784424,
+      "eval_runtime": 2.5776,
+      "eval_samples_per_second": 338.302,
+      "eval_steps_per_second": 2.716,
+      "step": 2635
+    },
+    {
+      "epoch": 6.0,
+      "grad_norm": Infinity,
+      "learning_rate": 1.1851340658998569e-05,
+      "loss": 0.6653,
+      "step": 3162
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.8325688073394495,
+      "eval_loss": 1.7560914754867554,
+      "eval_runtime": 2.5981,
+      "eval_samples_per_second": 335.631,
+      "eval_steps_per_second": 2.694,
+      "step": 3162
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 3689,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 7,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 48527917525620.0,
+  "train_batch_size": 128,
+  "trial_name": null,
+  "trial_params": {
+    "alpha": 0.1351029311037084,
+    "learning_rate": 8.264573854640023e-05,
+    "num_train_epochs": 7,
+    "temperature": 16
+  }
+}

run-3/checkpoint-3162/training_args.bin

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