End of training

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
+datasets:
+- imdb
 model-index:
 - name: distilbert-base-uncased-finetuned-imdb
   results: []
@@ -13,9 +15,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilbert-base-uncased-finetuned-imdb
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.2544
+- Loss: 2.4119
 
 ## Model description
 
@@ -35,25 +37,26 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 16
-- eval_batch_size: 16
+- train_batch_size: 64
+- eval_batch_size: 64
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - num_epochs: 3.0
+- mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 625  | 2.2733          |
-| 2.4263        | 2.0   | 1250 | 2.2270          |
-| 2.4263        | 3.0   | 1875 | 2.2056          |
+| 2.7024        | 1.0   | 157  | 2.4966          |
+| 2.5796        | 2.0   | 314  | 2.4282          |
+| 2.5355        | 3.0   | 471  | 2.4510          |
 
 
 ### Framework versions
 
-- Transformers 4.32.1
-- Pytorch 2.1.0
-- Datasets 2.12.0
-- Tokenizers 0.13.3
+- Transformers 4.35.0
+- Pytorch 2.1.0+cu118
+- Datasets 2.14.6
+- Tokenizers 0.14.1

config.json

@@ -19,6 +19,6 @@
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.32.1",
+  "transformers_version": "4.35.0",
   "vocab_size": 30522
 }

model.safetensors

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

runs/Nov03_00-06-29_939fad6a86e8/events.out.tfevents.1698970000.939fad6a86e8.708.0

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

runs/Nov03_00-06-29_939fad6a86e8/events.out.tfevents.1698970156.939fad6a86e8.708.1

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

tokenizer_config.json

@@ -1,4 +1,46 @@
 {
+  "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_lower_case": true,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2190fa110369c13857fac5d56d01b1ba01a7cb1620cac87ba94615bf08a30c19
-size 4536
+oid sha256:1f4a4748fbf24fc6c80bb5c716e0628c9fc84921c70a84be0359dc30888f7236
+size 4600