Spaces:
Runtime error
Runtime error
File size: 557 Bytes
1ef03e5 74618fc 1ef03e5 74618fc 1ef03e5 74618fc 1ef03e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
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") |