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.