Transfer Learning for Image Classification
Published:
Fine-tuned ResNet150 and ViT models on an image classification task. Developed a custom training pipeline on GCP comprising data augmentation and ensembling.
Tech Stack: Python, PyTorch, Torchvision, Vision Transformers, ResNet150, Google Cloud Platform (GCP)
Github repo here.