Hello / app.py
MihaiHuggingFace's picture
Update app.py
1ef03e5 verified
raw
history blame contribute delete
557 Bytes
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")