About Fine-Tuning#
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:
Create a Model Entity pointing to your base model checkpoint (stored as a FileSet).
Format a compatible dataset.
Create a customization job referencing the Model Entity.
Monitor the job until it completes.
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
Deploy the model using the Deployment Management Service.
Move on to Evaluate the output model.
Model Catalog#
Explore the model families and sizes supported by NVIDIA NeMo Customizer.
View the available Llama models in the model catalog.
View the available Llama Nemotron models from NVIDIA, including Nano and Super variants for efficient and advanced instruction tuning.
View the available Phi models from Microsoft, designed for strong reasoning capabilities with efficient deployment.
View the available GPT-OSS models supported for Full SFT customization.
View the available embedding models for question-answering and retrieval tasks.
For hardware compatibility, models requiring 80GB GPU memory work with A100(80GB), H100 and B200 GPUs. Refer to the model catalog pages for specific GPU requirements.
Task Guides#
Perform common fine-tuning tasks.
Create, list, view, and cancel customization jobs.
Create FileSets and Model Entities to prepare base models for customization.
Upload and manage datasets for training.
Tutorials#
Follow these tutorials to learn how to accomplish common fine-tuning tasks.
Learn how to format datasets for different model types.
Learn how to start a LoRA customization job using a custom dataset.
Learn how to start a SFT customization job using a custom dataset.
Learn how to check job metrics using MLFlow or Weights & Biases.
Learn how to optimize the token-per-GPU throughput for a LoRA optimization job.
References#
View the available hyperparameters and their valid options that you can set when creating a customization job.
View the OpenAPI specification for Customizer.
View troubleshooting tips for failed jobs.