← All projects

Machine Learning Engineer · April 2026

Paco's Puppy Problem

An image-classification study comparing CNNs, transfer learning, and a Vision Transformer across roughly 8,800 images.

Overview

What I built

This project focused on building and evaluating image-classification pipelines in PyTorch. I implemented a CNN and a Vision Transformer, explored transfer learning with different frozen-layer configurations, and designed a deeper regularized CNN using batch normalization, dropout, and weight decay. The experiments showed that transfer learning improved held-out generalization and that the CNN outperformed the ViT on this limited-data task.