raygx commited on
Commit
a77c556
1 Parent(s): 89593ba

Upload TFGPT2ForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +5 -5
  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 [raygx/distilGPT-Nepali](https://huggingface.co/raygx/distilGPT-Nepali) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 0.6596
18
- - Validation Loss: 0.6809
19
  - Epoch: 1
20
 
21
  ## Model description
@@ -35,15 +35,15 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': 1e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False, 'weight_decay_rate': 0.03}
39
  - training_precision: float32
40
 
41
  ### Training results
42
 
43
  | Train Loss | Validation Loss | Epoch |
44
  |:----------:|:---------------:|:-----:|
45
- | 0.8788 | 0.7572 | 0 |
46
- | 0.6596 | 0.6809 | 1 |
47
 
48
 
49
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [raygx/distilGPT-Nepali](https://huggingface.co/raygx/distilGPT-Nepali) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 0.6068
18
+ - Validation Loss: 0.6592
19
  - Epoch: 1
20
 
21
  ## Model description
 
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.04}
39
  - training_precision: float32
40
 
41
  ### Training results
42
 
43
  | Train Loss | Validation Loss | Epoch |
44
  |:----------:|:---------------:|:-----:|
45
+ | 0.8415 | 0.7254 | 0 |
46
+ | 0.6068 | 0.6592 | 1 |
47
 
48
 
49
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7cf295902ff41cd0fe53ad81486af836bd3c509e68586f9c7f3adbc99977219b
3
  size 480590728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:092cd73404d46b159572737e4cdfb64a5a5a1a42875bff2ae080b26f46535280
3
  size 480590728