Back to Browse

Fine-tune ConvNeXT for image classification [end-to-end tutorial]

6.6K views
Apr 21, 2022
47:25

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