Hugging Face Trainer Class

The Trainer class allows a user to train or fine-tune a model using a convenient function, rather than using native PyTorch.

When training, the Trainer will automatically use the GPU if one is present.

There are many options that can be set for the TrainingArguments (number of epochs, learning rate, save strategy, etc).

The Trainer will not automatically evaluate the LLM, so we will pass it an evaluation metric.

Previous
Next
RC Logo © 2025 The Rector and Visitors of the University of Virginia