GiantTreeG/german-jeopardy-mt5-large-256

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

german-jeopardy-mt5-large-256

This model is a fine-tuned version of google/mt5-large on the lmqg/qg_dequad dataset. It achieves the following results on the evaluation set:

• —Loss: 1.3943
• —Brevity Penalty: 0.9201
• —System Length: 19195
• —Reference Length: 20793
• —ROUGE-1: 43.56
• —ROUGE-2: 23.78
• —ROUGE-L: 41.81
• —ROUGE-Lsum: 41.80
• —Exact Match: 3.13
• —BLEU: 16.43
• —F1: 42.48

Model description

See google/mt5-large for the model architecture. The model was trained on a single NVIDIA RTX 3090 GPU with 24GB of VRAM.

Intended uses & limitations

This model can be used for question generation on German text.

Training and evaluation data

See lmqg/qg_dequad.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• —learning_rate: 0.0001
• —trainbatchsize: 1
• —evalbatchsize: 1
• —seed: 7
• —gradientaccumulationsteps: 256
• —totaltrainbatch_size: 256
• —optimizer: Adafactor
• —lrschedulertype: constant
• —num_epochs: 20

Training results

Framework versions

• —Transformers 4.32.1
• —Pytorch 2.1.0
• —Datasets 2.12.0
• —Tokenizers 0.13.3