Training in progress epoch 0

README.md

@@ -1,5 +1,6 @@
 ---
 license: apache-2.0
+base_model: bert-base-cased
 tags:
 - generated_from_keras_callback
 model-index:
@@ -14,9 +15,9 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0752
-- Validation Loss: 0.2672
-- Epoch: 9
+- Train Loss: 0.4561
+- Validation Loss: 0.3479
+- Epoch: 0
 
 ## Model description
 
@@ -35,28 +36,19 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 2310, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
+- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 5970, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
 - training_precision: mixed_float16
 
 ### Training results
 
 | Train Loss | Validation Loss | Epoch |
 |:----------:|:---------------:|:-----:|
-| 0.5102     | 0.3438          | 0     |
-| 0.3299     | 0.2654          | 1     |
-| 0.2570     | 0.2640          | 2     |
-| 0.2021     | 0.2583          | 3     |
-| 0.1612     | 0.2324          | 4     |
-| 0.1367     | 0.2319          | 5     |
-| 0.1129     | 0.2434          | 6     |
-| 0.0947     | 0.2651          | 7     |
-| 0.0816     | 0.2638          | 8     |
-| 0.0752     | 0.2672          | 9     |
+| 0.4561     | 0.3479          | 0     |
 
 
 ### Framework versions
 
-- Transformers 4.30.2
+- Transformers 4.31.0.dev0
 - TensorFlow 2.12.0
-- Datasets 2.13.1
+- Datasets 2.1.0
 - Tokenizers 0.13.3

config.json

@@ -10,24 +10,24 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "B-ORG",
-    "1": "O",
+    "0": "B-PER",
+    "1": "I-PER",
     "2": "I-ORG",
-    "3": "I-LOC",
-    "4": "I-PER",
-    "5": "B-PER",
-    "6": "B-LOC"
+    "3": "O",
+    "4": "B-LOC",
+    "5": "B-ORG",
+    "6": "I-LOC"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "B-LOC": 6,
-    "B-ORG": 0,
-    "B-PER": 5,
-    "I-LOC": 3,
+    "B-LOC": 4,
+    "B-ORG": 5,
+    "B-PER": 0,
+    "I-LOC": 6,
     "I-ORG": 2,
-    "I-PER": 4,
-    "O": 1
+    "I-PER": 1,
+    "O": 3
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
@@ -36,7 +36,7 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
-  "transformers_version": "4.30.2",
+  "transformers_version": "4.31.0.dev0",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 28996

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30ac60e5c5aa19bdd68a19a0971cc923521b9637aaacf8518fa5fc9b6990e613
+oid sha256:1caa9f9fbea2ec57e0b6278b4fbd2a3bc109029e04e4b801dcec143951a35bb5
 size 431173604