d2j666 commited on
Commit
eb9ff40
·
1 Parent(s): c02fecd

Upload model

Browse files
Files changed (2) hide show
  1. README.md +6 -6
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,8 +14,8 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 9.1610
18
- - Validation Loss: 8.9702
19
  - Epoch: 2
20
 
21
  ## Model description
@@ -36,15 +36,15 @@ More information needed
36
 
37
  The following hyperparameters were used during training:
38
  - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 5e-05, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 5e-05, 'decay_steps': -985, '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}
39
- - training_precision: mixed_float16
40
 
41
  ### Training results
42
 
43
  | Train Loss | Validation Loss | Epoch |
44
  |:----------:|:---------------:|:-----:|
45
- | 9.6117 | 9.5362 | 0 |
46
- | 9.4478 | 9.2722 | 1 |
47
- | 9.1610 | 8.9702 | 2 |
48
 
49
 
50
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 8.4905
18
+ - Validation Loss: 8.4397
19
  - Epoch: 2
20
 
21
  ## Model description
 
36
 
37
  The following hyperparameters were used during training:
38
  - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 5e-05, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 5e-05, 'decay_steps': -985, '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}
39
+ - training_precision: float32
40
 
41
  ### Training results
42
 
43
  | Train Loss | Validation Loss | Epoch |
44
  |:----------:|:---------------:|:-----:|
45
+ | 8.8734 | 8.7298 | 0 |
46
+ | 8.6494 | 8.5595 | 1 |
47
+ | 8.4905 | 8.4397 | 2 |
48
 
49
 
50
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8bece7985fb903896dca0a1aa5e8b50420716d602e227e6a0a6145d01370e403
3
  size 385107024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86e318f6b941db2b4c99bee64ed4f900d3cfb8fb2bb2f4aafca816c7ff417fc9
3
  size 385107024