Custom train

#1
by Ashv27 - opened

How to custom train on own data?

Hi,

To fine-tune this model I'd recommend checking out the official example notebook: https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/image_classification.ipynb.

Tip: make sure to set ignore_mismatched_sizes=True in the from_pretrained method to replace the head of the model.

Sign up or log in to comment