apple / ml-cvnets

CVNets: A library for training computer vision networks
https://apple.github.io/ml-cvnets
Other
1.76k stars 225 forks source link

Update README-classification-tutorial.md #83

Open bolotyuh opened 1 year ago

bolotyuh commented 1 year ago

When you want fine-tuning model on a custom dataset with another N of classes you have to pass argument --model.resume-exclude-scopes classifier otherwise, we will get an error about the mismatch classifier layer.