Hi guys!
In this video, I show how to fine-tune ConvNeXT, a state-of-the-art model by Facebook AI for image classification on a custom dataset. I leverage the new 🤗 ImageFolder feature, and train the model in native PyTorch.
Timestamps:
0:55 Intro to ConvNeXT
1:35 Loading a custom dataset using ImageFolder
6:40 Push dataset to the 🤗 hub
8:57 Process dataset
18:10 Define model
21:20 Prepare training of the model
34:30 Start training!
36:46 Push model to the hub
38:15 Performing inference (with widget)
44:20 Pipeline API
Colab notebook: https://github.com/NielsRogge/Transformers-Tutorials/blob/master/ConvNeXT/Fine_tune_ConvNeXT_for_image_classification.ipynb
Download
0 formats
No download links available.
Fine-tune ConvNeXT for image classification [end-to-end tutorial] | NatokHD