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") |