Oct. 5, 2023, 10:32 a.m. | Markus Stoll

DEV Community dev.to




How I Created an Animation Of the Embeddings During Fine-Tuning





Using Cleanlab, PCA, and Procrustes to visualize ViT fine-tuning on CIFAR-10


In the field of machine learning, Vision Transformers (ViT) are a type of model used for image classification. Unlike traditional convolutional neural networks, ViTs use the transformer architecture, which was originally designed for natural language processing tasks, to process images. Fine-tuning these models, for optimal performance can be a complex process.


In a previous article, I used an …

ai animation architecture cifar-10 classification cleanlab convolutional neural networks embeddings fine-tuning github image machine machine learning networks neural networks python transformer transformer architecture transformers tutorial type vision vision transformers vit

Founding AI Engineer, Agents

@ Occam AI | New York

AI Engineer Intern, Agents

@ Occam AI | US

AI Research Scientist

@ Vara | Berlin, Germany and Remote

Data Architect

@ University of Texas at Austin | Austin, TX

Data ETL Engineer

@ University of Texas at Austin | Austin, TX

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne