Pass the training arguments to [Trainer] along with the model, dataset, tokenizer, data collator, and compute_metrics function.