
Finetuning
Generates code that fine-tunes a base model using SageMaker serverless training jobs.
Install
npx skills add awslabs/agent-plugins --skill finetuning -g -a claude-code -y- Repository
- awslabs/agent-plugins933
- Category
- AI and agents
- Updated
- 20d ago
- License
- Apache-2.0
- Context cost
- Mediumabout 2,222 tokens
- Runs scripts
- Yes, review before use
What it does
Generates code that fine-tunes a base model using SageMaker serverless training jobs. Use when the user says "start training", "fine-tune my model", "I'm ready to train", or when the plan reaches the finetuning step. Supports SFT, DPO, RLVR, and RLAIF trainers, including RLVR Lambda reward function and RLAIF custom prompt creation.
Prerequisites
Before starting this workflow, verify:
-
A
use_case_spec.mdfile exists- If missing: Activate the
use-case-specificationskill first, then resume - DON'T EVER offer to create a use case spec without activating the use-case-specification skill.
- If missing: Activate the
-
A fine-tuning technique (SFT, DPO, RLVR, RLAIF, or CPT/RFT (for Nova)) and base model have already been selected
- If missing: Activate the
model-selectionand/orfinetuning-techniqueskills to collect what's missing, then resume - Don't make recommendations on the spot. You MUST activate the appropriate skill.
- If missing: Activate the
-
A base model name available on SageMakerHub has been identified
- If missing: Activate the
model-selectionskill to get it - Important: Only use the model name that
model-selectionretrieves, as it may differ from other commonly used names for the same model
- If missing: Activate the
-
The SDK environment has been verified (SDK version, region, execution role)
- If not done: Activate the
sdk-getting-startedskill first, then resume
- If not done: Activate the
-
A training dataset uploaded to a bucket in the environment's default region.
- If not met: Help the user upload the dataset to the correct S3
Critical Rules
Code Generation Rules
- ✅ Use EXACTLY the imports shown in each code template
- ❌ Do NOT add additional imports even if they seem helpful
- ❌ Do NOT create variables before they're needed in that section
- 📋 Copy the code structure precisely - no improvisation
- 🎯 Follow the minimal code principle strictly
- ✅ When writing code, make sure the indentation and f strings are correct

