peterhsu commited on
Commit
1440d9e
1 Parent(s): 0bc16fe
README.md CHANGED
@@ -3,20 +3,18 @@ license: apache-2.0
3
  tags:
4
  - generated_from_keras_callback
5
  model-index:
6
- - name: peterhsu/tf-distilbert-base-uncased-finetuned-imdb
7
  results: []
8
  ---
9
 
10
  <!-- This model card has been generated automatically according to the information Keras had access to. You should
11
  probably proofread and complete it, then remove this comment. -->
12
 
13
- # peterhsu/tf-distilbert-base-uncased-finetuned-imdb
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 2.8673
18
- - Validation Loss: 2.5941
19
- - Epoch: 0
20
 
21
  ## Model description
22
 
@@ -35,14 +33,11 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 2e-05, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': -688, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, '__passive_serialization__': True}, 'warmup_steps': 1000, 'power': 1.0, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
39
  - training_precision: mixed_float16
40
 
41
  ### Training results
42
 
43
- | Train Loss | Validation Loss | Epoch |
44
- |:----------:|:---------------:|:-----:|
45
- | 2.8673 | 2.5941 | 0 |
46
 
47
 
48
  ### Framework versions
 
3
  tags:
4
  - generated_from_keras_callback
5
  model-index:
6
+ - name: tf-distilbert-base-uncased-finetuned-imdb
7
  results: []
8
  ---
9
 
10
  <!-- This model card has been generated automatically according to the information Keras had access to. You should
11
  probably proofread and complete it, then remove this comment. -->
12
 
13
+ # tf-distilbert-base-uncased-finetuned-imdb
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) 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: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 2e-05, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': -688, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, '__passive_serialization__': True}, 'warmup_steps': 1000, 'power': 1.0, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
37
  - training_precision: mixed_float16
38
 
39
  ### Training results
40
 
 
 
 
41
 
42
 
43
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:14794ee2df01546df54d728c40ac35a47b4de64c5708f5839270c299e0bc1162
3
  size 363423680
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:015ce8eade54d8b998fbf99c0ff5176fe13fe0c91d710b177a13edc774f2808f
3
  size 363423680
train/events.out.tfevents.1645023038.1aff28ad178a.2391.0.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c65743bdd5bcb9ddc87f3434995b30623d941e1ff76e870ad657c4fa9925e005
3
+ size 1531540
validation/events.out.tfevents.1645023392.1aff28ad178a.2391.1.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e461607d902c326b4fa775e8c29a5dc16708c16f9f78cefe111bcee79789eeed
3
+ size 194