🔭 Train a Vision Model with 🤗 Transformers
Published on Feb 3, 2023
Forked from tobyho/TorchVisionWeights & Biases is a machine learning experiment tracking, model checkpointing and data visualisation tool used by over 200,000 ML practitioners across the world.
You needs to turn on the Replit GPU instance for this script to train
This script leverages the computer vision components in the hugging face ecosystem to train a computer vision model to classify different types of food. You can swap out the dataset and model easily to customise it and make it your own!
Created with ❤️ by Weights & Biases
Sign up to Weights & Biases here to get started!
Loading comments...