Spaces:
Runtime error
Runtime error
| from transformers import AutoModelForSeq2SeqLM | |
| model_name = "facebook/m2m100_418M" | |
| model = AutoModelForSeq2SeqLM.from_pretrained(model_name) | |
| from transformers import TrainingArguments | |
| training_args = TrainingArguments( | |
| output_dir="./results", | |
| num_train_epochs=10, | |
| per_device_train_batch_size=8, | |
| save_steps=1000, | |
| ) | |
| from transformers import Trainer | |
| trainer = Trainer( | |
| model=model, | |
| args=training_args, | |
| train_dataset=train_dataset, | |
| eval_dataset=eval_dataset, | |
| ) | |
| trainer.train() | |
| trainer.save_model("m2m100_en_paa_finetuned") |