Training in progress epoch 0

README.md

@@ -14,9 +14,9 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: nan
-- Validation Loss: nan
-- Epoch: 7
+- Train Loss: 5.5637
+- Validation Loss: 3.0045
+- Epoch: 0
 
 ## Model description
 
@@ -35,21 +35,14 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 0.0004, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 0.0004, 'decay_steps': 12532, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, '__passive_serialization__': True}, 'warmup_steps': 100, '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}
+- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 0.0003, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 0.0003, 'decay_steps': 1924, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, '__passive_serialization__': True}, 'warmup_steps': 50, '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.001}
 - training_precision: mixed_float16
 
 ### Training results
 
 | Train Loss | Validation Loss | Epoch |
 |:----------:|:---------------:|:-----:|
-| nan        | nan             | 0     |
-| nan        | nan             | 1     |
-| nan        | nan             | 2     |
-| nan        | nan             | 3     |
-| nan        | nan             | 4     |
-| nan        | nan             | 5     |
-| nan        | nan             | 6     |
-| nan        | nan             | 7     |
+| 5.5637     | 3.0045          | 0     |
 
 
 ### Framework versions

config.json

@@ -18,6 +18,7 @@
   "length_penalty": 0.6,
   "max_length": 128,
   "model_type": "mt5",
+  "no_repeat_ngram_size": 2,
   "num_beams": 15,
   "num_decoder_layers": 8,
   "num_heads": 6,

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0f6654ed46c303812a0171c599f72e90593c2f5bb7e88b5ae56bfa819d15b568
-size 2225560376
+oid sha256:cf55598e68a86bdf1cb74717b623753a8f77ca6ff11d067532a659e020c40ace
+size 2225556280