About Fine-Tuning

View as Markdown

Learn how to fine-tune models by making requests to NVIDIA NeMo Customizer through the API. Fine-tuned models you have created can be deployed using NVIDIA NIMs.

Fine-Tuning Workflow

At a high level, the fine-tuning workflow consists of the following steps:

  1. Create a Model Entity pointing to your base model checkpoint (stored as a FileSet).
  2. Format a compatible dataset.
  3. Create a customization job referencing the Model Entity.
  4. Monitor the job until it completes.
  5. The customization job automatically creates either:
  • LoRA jobs: An adapter attached to the original Model Entity
  • Full fine-tuning jobs: A new Model Entity with the customized weights
  1. Deploy the model using the Deployment Management Service.
  2. Move on to Evaluate the output model.

Model Catalog

Explore the model families and sizes supported by NVIDIA NeMo Customizer.

Task Guides

Perform common fine-tuning tasks.


Tutorials

Follow these tutorials to learn how to accomplish common fine-tuning tasks.


References