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