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.