train.py

from transformers import Trainer

trainer = Trainer(model=model, args=training_args,
                  compute_metrics=compute_metrics,
                  train_dataset=offensive_encoded["train"],
                  eval_dataset=offensive_encoded["validation"],
                  tokenizer=tokenizer)
trainer.train();