Upload TFBartForConditionalGeneration

README.md

@@ -15,9 +15,9 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [facebook/bart-large](https://huggingface.co/facebook/bart-large) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.3501
-- Validation Loss: 0.6153
-- Epoch: 7
+- Train Loss: 0.5063
+- Validation Loss: 0.6842
+- Epoch: 4
 
 ## Model description
 
@@ -36,26 +36,23 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- 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': 1e-05}
+- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.001}
 - training_precision: float32
 
 ### Training results
 
 | Train Loss | Validation Loss | Epoch |
 |:----------:|:---------------:|:-----:|
-| 1.3410     | 0.9278          | 0     |
-| 0.8642     | 0.8051          | 1     |
-| 0.7179     | 0.6812          | 2     |
-| 0.6052     | 0.7129          | 3     |
-| 0.5071     | 0.6385          | 4     |
-| 0.4461     | 0.6174          | 5     |
-| 0.3939     | 0.6276          | 6     |
-| 0.3501     | 0.6153          | 7     |
+| 1.4675     | 0.9956          | 0     |
+| 0.9108     | 0.8082          | 1     |
+| 0.7168     | 0.7359          | 2     |
+| 0.5947     | 1.1556          | 3     |
+| 0.5063     | 0.6842          | 4     |
 
 
 ### Framework versions
 
-- Transformers 4.33.1
+- Transformers 4.33.2
 - TensorFlow 2.13.0
 - Datasets 2.14.5
 - Tokenizers 0.13.3

config.json

@@ -67,7 +67,7 @@
       "num_beams": 6
     }
   },
-  "transformers_version": "4.33.1",
+  "transformers_version": "4.33.2",
   "use_cache": true,
   "vocab_size": 50265
 }

generation_config.json

@@ -9,5 +9,5 @@
   "no_repeat_ngram_size": 3,
   "num_beams": 4,
   "pad_token_id": 1,
-  "transformers_version": "4.33.1"
+  "transformers_version": "4.33.2"
 }

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8f3379c1b5233daa4d9a43d1682edd3c9dbf1e96cf5c6bd6644a171d9b93b1d0
+oid sha256:9e5f71818b4531faf0a1c52c9e4ad71b6c8f82e53c0f13654af2fca59b980212
 size 1625925412