Tiny VGG Model in PyTorch This repository contains a replication of the Tiny VGG model implemented in PyTorch, designed for image classification on custom datasets. Replicated https://poloclub.github.io/cnn-explainer/ this model in pytorch
Requirements Python 3.x PyTorch