add model
Browse files
README.md
CHANGED
@@ -3,20 +3,18 @@ license: apache-2.0
|
|
3 |
tags:
|
4 |
- generated_from_keras_callback
|
5 |
model-index:
|
6 |
-
- name:
|
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 |
-
#
|
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 |
-
- 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: {'
|
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:
|
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
|