← Back to Course

Assignment 3: Classification and Instruction Fine-Tuning

Fine-Tune Your AI Models!

Implement classification head fine-tuning (Chapter 6) and instruction fine-tuning (Chapter 7) using GPT-2

Assignment Overview

This assignment has two main components:

  • classification_model.py - Classification Model Implementation (60 points)
  • instruction_model.py - Instruction Fine-Tuning Implementation (40 points)

Complete all TODO sections in the starter code. The implementations should follow the hints and maintain the original function signatures.

↑
Drop your file here or click to browse
classification_model.py • Max 5MB
✓
Complete implementation of classification head fine-tuning (60 points)
↑
Drop your file here or click to browse
instruction_model.py • Max 5MB
✓
Complete implementation of instruction fine-tuning (40 points)
Submission Requirements:
  • Use the exact file names shown above (case-sensitive)
  • Complete all TODO implementations in the starter code
  • Maintain original function signatures and class names
  • Each file should not exceed 5MB