Ramuvannela commited on
Commit
2f00dfa
1 Parent(s): 8bf0a34

End of training

Browse files
Files changed (3) hide show
  1. README.md +2 -11
  2. config.json +2 -2
  3. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,9 +14,7 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [gchhablani/bert-base-cased-finetuned-sst2](https://huggingface.co/gchhablani/bert-base-cased-finetuned-sst2) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 0.0285
18
- - Validation Loss: 0.3471
19
- - Epoch: 4
20
 
21
  ## Model description
22
 
@@ -35,18 +33,11 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False, 'weight_decay_rate': 0.01}
39
  - training_precision: float32
40
 
41
  ### Training results
42
 
43
- | Train Loss | Validation Loss | Epoch |
44
- |:----------:|:---------------:|:-----:|
45
- | 0.0710 | 0.2416 | 0 |
46
- | 0.0578 | 0.2365 | 1 |
47
- | 0.0432 | 0.3337 | 2 |
48
- | 0.0343 | 0.3451 | 3 |
49
- | 0.0285 | 0.3471 | 4 |
50
 
51
 
52
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [gchhablani/bert-base-cased-finetuned-sst2](https://huggingface.co/gchhablani/bert-base-cased-finetuned-sst2) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+
 
 
18
 
19
  ## Model description
20
 
 
33
  ### Training hyperparameters
34
 
35
  The following hyperparameters were used during training:
36
+ - optimizer: None
37
  - training_precision: float32
38
 
39
  ### Training results
40
 
 
 
 
 
 
 
 
41
 
42
 
43
  ### Framework versions
config.json CHANGED
@@ -17,8 +17,8 @@
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
- "negative": "0",
21
- "positive": "1"
22
  },
23
  "layer_norm_eps": 1e-12,
24
  "max_position_embeddings": 512,
 
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
+ "negative": 0,
21
+ "positive": 1
22
  },
23
  "layer_norm_eps": 1e-12,
24
  "max_position_embeddings": 512,
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5006e280e0070aae7e4e48a8f57327dffe1038f00621dc466ee3de09962a7496
3
  size 433535320
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44eb841786f55d6e34c1b2597b4ef45b6b6348e5ac09f93cb89e9925277d3277
3
  size 433535320