End of training

README.md

@@ -1,6 +1,5 @@
 ---
-license: mit
-base_model: roberta-base
+base_model: huawei-noah/TinyBERT_General_4L_312D
 tags:
 - generated_from_trainer
 metrics:
@@ -15,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # NLP_Capstone
 
-This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
+This model is a fine-tuned version of [huawei-noah/TinyBERT_General_4L_312D](https://huggingface.co/huawei-noah/TinyBERT_General_4L_312D) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4545
-- Accuracy: 0.9074
+- Loss: 0.5352
+- Accuracy: 0.7654
 
 ## Model description
 
@@ -37,7 +36,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 5e-05
+- learning_rate: 2e-05
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42

config.json

@@ -1,28 +1,30 @@
 {
-  "_name_or_path": "roberta-base",
+  "_name_or_path": "huawei-noah/TinyBERT_General_4L_312D",
   "architectures": [
-    "RobertaForSequenceClassification"
+    "BertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
-  "bos_token_id": 0,
+  "cell": {},
   "classifier_dropout": null,
-  "eos_token_id": 2,
+  "emb_size": 312,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
-  "hidden_size": 768,
+  "hidden_size": 312,
   "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "layer_norm_eps": 1e-05,
-  "max_position_embeddings": 514,
-  "model_type": "roberta",
+  "intermediate_size": 1200,
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
   "num_attention_heads": 12,
-  "num_hidden_layers": 12,
-  "pad_token_id": 1,
+  "num_hidden_layers": 4,
+  "pad_token_id": 0,
   "position_embedding_type": "absolute",
+  "pre_trained": "",
   "problem_type": "single_label_classification",
+  "structure": [],
   "torch_dtype": "float32",
   "transformers_version": "4.34.1",
-  "type_vocab_size": 1,
+  "type_vocab_size": 2,
   "use_cache": true,
-  "vocab_size": 50265
+  "vocab_size": 30522
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9f5bdec7ef595fb1162ce437a43e2ee9c3ab3eec3ab55b1a5588950404e38dbf
-size 498658094
+oid sha256:d38ac6260d27ff760bd756900296baf8a88997973f4ca5b50d9f64ad91defa79
+size 57427209

special_tokens_map.json

@@ -1,15 +1,7 @@
 {
-  "bos_token": "<s>",
-  "cls_token": "<s>",
-  "eos_token": "</s>",
-  "mask_token": {
-    "content": "<mask>",
-    "lstrip": true,
-    "normalized": false,
-    "rstrip": false,
-    "single_word": false
-  },
-  "pad_token": "<pad>",
-  "sep_token": "</s>",
-  "unk_token": "<unk>"
+  "cls_token": "[CLS]",
+  "mask_token": "[MASK]",
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "unk_token": "[UNK]"
 }

tokenizer_config.json

@@ -1,57 +1,57 @@
 {
-  "add_prefix_space": false,
   "added_tokens_decoder": {
     "0": {
-      "content": "<s>",
+      "content": "[PAD]",
       "lstrip": false,
-      "normalized": true,
+      "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "1": {
-      "content": "<pad>",
+    "100": {
+      "content": "[UNK]",
       "lstrip": false,
-      "normalized": true,
+      "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "2": {
-      "content": "</s>",
+    "101": {
+      "content": "[CLS]",
       "lstrip": false,
-      "normalized": true,
+      "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "3": {
-      "content": "<unk>",
+    "102": {
+      "content": "[SEP]",
       "lstrip": false,
-      "normalized": true,
+      "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "50264": {
-      "content": "<mask>",
-      "lstrip": true,
+    "103": {
+      "content": "[MASK]",
+      "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     }
   },
-  "bos_token": "<s>",
   "clean_up_tokenization_spaces": true,
-  "cls_token": "<s>",
-  "eos_token": "</s>",
-  "errors": "replace",
-  "mask_token": "<mask>",
-  "model_max_length": 512,
-  "pad_token": "<pad>",
-  "sep_token": "</s>",
-  "tokenizer_class": "RobertaTokenizer",
-  "trim_offsets": true,
-  "unk_token": "<unk>"
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "model_max_length": 1000000000000000019884624838656,
+  "never_split": null,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "unk_token": "[UNK]"
 }

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4f1d22d62781e7120b84cec426bb173a6e9c9d4ad3edb28c6a781fce09fbf6d4
+oid sha256:9823ddfb06613a12632e86f9b83d11ec2b2f6cb6e5b317af8c18d5bfc2821947
 size 4472