manuu01 commited on
Commit
06ebc45
1 Parent(s): 1b6daa0

Upload TFBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +12 -6
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -13,7 +13,11 @@ probably proofread and complete it, then remove this comment. -->
13
 
14
  This model was trained from scratch on an unknown dataset.
15
  It achieves the following results on the evaluation set:
16
-
 
 
 
 
17
 
18
  ## Model description
19
 
@@ -32,14 +36,16 @@ More information needed
32
  ### Training hyperparameters
33
 
34
  The following hyperparameters were used during training:
35
- - optimizer: None
36
  - training_precision: float32
37
 
38
  ### Training results
39
- It achieved the following accuracy during training (on validation sets):
40
- SNLI: 87,90%
41
- MNLI: 82.27%
42
- ANLI_r3: 44.83%
 
 
43
 
44
 
45
  ### Framework versions
 
13
 
14
  This model was trained from scratch on an unknown dataset.
15
  It achieves the following results on the evaluation set:
16
+ - Train Loss: 0.1618
17
+ - Train Accuracy: 0.9368
18
+ - Validation Loss: 0.2226
19
+ - Validation Accuracy: 0.9102
20
+ - Epoch: 2
21
 
22
  ## Model description
23
 
 
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
+ - optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 2214, 'end_learning_rate': 1e-05, 'power': 1.0, 'cycle': False, 'name': None}}, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
40
  - training_precision: float32
41
 
42
  ### Training results
43
+
44
+ | Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
45
+ |:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
46
+ | 0.3017 | 0.8710 | 0.2638 | 0.8843 | 0 |
47
+ | 0.1989 | 0.9198 | 0.2287 | 0.9139 | 1 |
48
+ | 0.1618 | 0.9368 | 0.2226 | 0.9102 | 2 |
49
 
50
 
51
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e567ca24afc2498079d5887e4bbc181128eaea99603b4105f613397d1b776076
3
  size 51152184
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfe51073c1ae69cb2dba99c9c5ea26218735d273c4875b90eebab6c839550d67
3
  size 51152184