Upload TFT5ForConditionalGeneration

README.md

@@ -14,9 +14,9 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 3.8958
-- Validation Loss: 3.5264
-- Train Rougel: tf.Tensor(0.13228828, shape=(), dtype=float32)
+- Train Loss: 3.8679
+- Validation Loss: 3.5523
+- Train Rougel: tf.Tensor(0.11994212, shape=(), dtype=float32)
 - Epoch: 0
 
 ## Model description
@@ -36,19 +36,19 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': False, 'is_legacy_optimizer': False, 'learning_rate': 2e-05, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
+- optimizer: {'name': 'Adam', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
 - training_precision: float32
 
 ### Training results
 
 | Train Loss | Validation Loss | Train Rougel                                   | Epoch |
 |:----------:|:---------------:|:----------------------------------------------:|:-----:|
-| 3.8958     | 3.5264          | tf.Tensor(0.13228828, shape=(), dtype=float32) | 0     |
+| 3.8679     | 3.5523          | tf.Tensor(0.11994212, shape=(), dtype=float32) | 0     |
 
 
 ### Framework versions
 
-- Transformers 4.20.0
-- TensorFlow 2.13.0
-- Datasets 2.14.4
+- Transformers 4.26.1
+- TensorFlow 2.10.1
+- Datasets 2.13.1
 - Tokenizers 0.12.1

config.json

@@ -60,7 +60,7 @@
       "prefix": "translate English to Romanian: "
     }
   },
-  "transformers_version": "4.20.0",
+  "transformers_version": "4.26.1",
   "use_cache": true,
   "vocab_size": 32128
 }

generation_config.json

@@ -0,0 +1,13 @@
+{
+  "_from_model_config": true,
+  "decoder_start_token_id": 0,
+  "early_stopping": true,
+  "eos_token_id": 1,
+  "length_penalty": 2.0,
+  "max_length": 200,
+  "min_length": 30,
+  "no_repeat_ngram_size": 3,
+  "num_beams": 4,
+  "pad_token_id": 0,
+  "transformers_version": "4.26.1"
+}

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5bc56c3d9f36c7dc846c294bf88c4304fe2b6af29f722f3ba8b635bb41ce70ef
-size 242301696
+oid sha256:4e5c73b02d0ffc722ade34ff3f9abd341d9856bbc11caecb4c99d1ea9c5be19c
+size 373902664