Slug Classification with Vision Transformer (ViT)

This repository contains implementation of a Vision Transformer (ViT) model for slug classification. The project uses PyTorch for implementing the ViT architecture and training the model on slug image data.

Project Structure

WIP

• slug-classifier/: Code for training and evaluating the model