All Katas - ML Katas

by: newest upvotes saves

Transfer Learning with a Pre-trained Model

this month medium (<1 hr) | cnn transfer learning resnet fine-tuning

Fine-tune a **pre-trained model** (e.g., `resnet18` from `torchvision.models`) on a new, small image classification dataset (e.g., `CIFAR-10`). You'll need to freeze the weights of the initial...