Hugging Face's logo

aish31
/
Prompt_genre

Text2Text Generation
Transformers
TensorFlow
bart
generated_from_keras_callback
Inference Endpoints
Model card Files Community
aish31 commited on
Commit
6c85abd
·
1 Parent(s): 95eca41

Upload TFBartForConditionalGeneration

Browse files
Files changed (2) hide show
  1. README.md +10 -10
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -15,8 +15,8 @@ probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [facebook/bart-large](https://huggingface.co/facebook/bart-large) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.3563
19
- - Validation Loss: 0.6763
20
  - Epoch: 7
21
 
22
  ## Model description
@@ -43,14 +43,14 @@ The following hyperparameters were used during training:
43
 
44
  | Train Loss | Validation Loss | Epoch |
45
  |:----------:|:---------------:|:-----:|
46
- | 1.3569 | 0.9766 | 0 |
47
- | 0.9065 | 0.8187 | 1 |
48
- | 0.7160 | 0.7519 | 2 |
49
- | 0.5987 | 0.6857 | 3 |
50
- | 0.5152 | 0.6609 | 4 |
51
- | 0.4508 | 0.6720 | 5 |
52
- | 0.4014 | 0.6485 | 6 |
53
- | 0.3563 | 0.6763 | 7 |
54
 
55
 
56
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [facebook/bart-large](https://huggingface.co/facebook/bart-large) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 0.3501
19
+ - Validation Loss: 0.6153
20
  - Epoch: 7
21
 
22
  ## Model description
 
43
 
44
  | Train Loss | Validation Loss | Epoch |
45
  |:----------:|:---------------:|:-----:|
46
+ | 1.3410 | 0.9278 | 0 |
47
+ | 0.8642 | 0.8051 | 1 |
48
+ | 0.7179 | 0.6812 | 2 |
49
+ | 0.6052 | 0.7129 | 3 |
50
+ | 0.5071 | 0.6385 | 4 |
51
+ | 0.4461 | 0.6174 | 5 |
52
+ | 0.3939 | 0.6276 | 6 |
53
+ | 0.3501 | 0.6153 | 7 |
54
 
55
 
56
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1a160c87bcbc04d4c5c091af1dca81992bc843d08594765677826d6b3e8dca0e
3
  size 1625925412
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f3379c1b5233daa4d9a43d1682edd3c9dbf1e96cf5c6bd6644a171d9b93b1d0
3
  size 1625925412