--- license: apache-2.0 base_model: google/mt5-base tags: - generated_from_trainer metrics: - rouge - sacrebleu model-index: - name: mT5-TextSimp-LT-BatchSize2-lr1e-4 results: [] --- # mT5-TextSimp-LT-BatchSize2-lr1e-4 This model is a fine-tuned version of [google/mt5-base](https://huggingface.co/google/mt5-base) on the None dataset. It achieves the following results on the evaluation set: - Loss: 0.0672 - Rouge1: 0.7548 - Rouge2: 0.5989 - Rougel: 0.7509 - Sacrebleu: 49.0373 - Gen Len: 38.0501 ## Model description More information needed ## Intended uses & limitations More information needed ## Training and evaluation data More information needed ## Training procedure ### Training hyperparameters The following hyperparameters were used during training: - learning_rate: 0.0001 - train_batch_size: 2 - eval_batch_size: 2 - seed: 42 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 - lr_scheduler_type: linear - lr_scheduler_warmup_steps: 500 - num_epochs: 8 ### Training results | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Sacrebleu | Gen Len | |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:| | 25.6783 | 0.24 | 200 | 16.0497 | 0.0109 | 0.0005 | 0.0107 | 0.0029 | 512.0 | | 1.9593 | 0.48 | 400 | 0.7780 | 0.014 | 0.0005 | 0.0136 | 0.0146 | 42.685 | | 0.2778 | 0.72 | 600 | 0.1429 | 0.4924 | 0.3128 | 0.4803 | 20.3057 | 38.0382 | | 0.1325 | 0.96 | 800 | 0.1039 | 0.6193 | 0.4369 | 0.6098 | 33.687 | 38.0501 | | 0.1702 | 1.2 | 1000 | 0.0958 | 0.6697 | 0.5016 | 0.6613 | 38.0391 | 38.0501 | | 0.13 | 1.44 | 1200 | 0.0880 | 0.6737 | 0.5051 | 0.6644 | 38.62 | 38.0501 | | 0.1086 | 1.67 | 1400 | 0.0839 | 0.6964 | 0.5326 | 0.6884 | 40.9056 | 38.0501 | | 0.0716 | 1.91 | 1600 | 0.0859 | 0.6933 | 0.5298 | 0.6862 | 40.7158 | 38.0501 | | 0.1135 | 2.15 | 1800 | 0.0820 | 0.7017 | 0.5366 | 0.6936 | 40.7484 | 38.0501 | | 0.0997 | 2.39 | 2000 | 0.0814 | 0.7011 | 0.5351 | 0.6945 | 41.1948 | 38.0501 | | 0.0996 | 2.63 | 2200 | 0.0774 | 0.7103 | 0.5522 | 0.7049 | 42.5756 | 38.0501 | | 1.1379 | 2.87 | 2400 | 0.0763 | 0.7211 | 0.5556 | 0.7152 | 43.2411 | 38.0501 | | 0.0594 | 3.11 | 2600 | 0.0776 | 0.7261 | 0.5647 | 0.7201 | 44.2205 | 38.0501 | | 0.0763 | 3.35 | 2800 | 0.0736 | 0.7309 | 0.5709 | 0.7251 | 45.2825 | 38.0501 | | 0.1641 | 3.59 | 3000 | 0.0722 | 0.7297 | 0.5685 | 0.7242 | 44.9001 | 38.0501 | | 0.1085 | 3.83 | 3200 | 0.0703 | 0.7377 | 0.5793 | 0.7319 | 45.7504 | 38.0501 | | 0.0573 | 4.07 | 3400 | 0.0719 | 0.7393 | 0.5796 | 0.7335 | 45.86 | 38.0501 | | 0.1149 | 4.31 | 3600 | 0.0705 | 0.7415 | 0.5787 | 0.7365 | 46.2652 | 38.0501 | | 0.0843 | 4.55 | 3800 | 0.0703 | 0.7385 | 0.5754 | 0.7326 | 46.5292 | 38.0501 | | 0.0658 | 4.78 | 4000 | 0.0705 | 0.7437 | 0.5855 | 0.7384 | 46.864 | 38.0501 | | 0.0676 | 5.02 | 4200 | 0.0694 | 0.7437 | 0.584 | 0.7384 | 47.1268 | 38.0501 | | 0.0657 | 5.26 | 4400 | 0.0711 | 0.7473 | 0.5913 | 0.7432 | 47.4413 | 38.0501 | | 0.0679 | 5.5 | 4600 | 0.0702 | 0.7496 | 0.5908 | 0.7446 | 47.8281 | 38.0501 | | 0.0664 | 5.74 | 4800 | 0.0671 | 0.7511 | 0.5929 | 0.7463 | 47.7693 | 38.0501 | | 0.0446 | 5.98 | 5000 | 0.0685 | 0.7533 | 0.5932 | 0.7478 | 48.032 | 38.0501 | | 0.0732 | 6.22 | 5200 | 0.0678 | 0.7523 | 0.5948 | 0.7472 | 48.3467 | 38.0501 | | 0.0706 | 6.46 | 5400 | 0.0672 | 0.755 | 0.5983 | 0.7507 | 48.6158 | 38.0501 | | 0.051 | 6.7 | 5600 | 0.0674 | 0.7523 | 0.5961 | 0.7478 | 48.4828 | 38.0501 | | 0.067 | 6.94 | 5800 | 0.0681 | 0.7532 | 0.5978 | 0.7492 | 48.7253 | 38.0501 | | 0.075 | 7.18 | 6000 | 0.0684 | 0.7534 | 0.5969 | 0.7492 | 48.7053 | 38.0501 | | 0.1323 | 7.42 | 6200 | 0.0671 | 0.755 | 0.5991 | 0.7511 | 48.9922 | 38.0501 | | 0.0383 | 7.66 | 6400 | 0.0671 | 0.7551 | 0.5994 | 0.7511 | 49.0028 | 38.0501 | | 0.0599 | 7.89 | 6600 | 0.0672 | 0.7548 | 0.5989 | 0.7509 | 49.0373 | 38.0501 | ### Framework versions - Transformers 4.33.0 - Pytorch 2.1.2+cu121 - Datasets 2.14.4 - Tokenizers 0.13.3