johko / computer-vision-course

This repo is the homebase of a community driven course on Computer Vision with Neural Networks. Feel free to join us on the Hugging Face discord: hf.co/join/discord
MIT License
372 stars 123 forks source link

Adding Knowledge Distillation (Image Classification) #111

Closed asusevski closed 6 months ago

review-notebook-app[bot] commented 6 months ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

review-notebook-app[bot] commented 6 months ago

View / edit / reply to this conversation on ReviewNB

MKhalusova commented on 2023-12-11T15:23:35Z ----------------------------------------------------------------

You can use AutoModelForImageClassification here.


review-notebook-app[bot] commented 6 months ago

View / edit / reply to this conversation on ReviewNB

MKhalusova commented on 2023-12-11T15:23:36Z ----------------------------------------------------------------

You can push the model to Hub to show how to share models with the community. (In this case, you'll also need to log in and possibly change the output_dir as it will be used as a repo name on Hub).


asusevski commented 6 months ago

Sounds good, I'm currently away for Christmas but I'll address these minor changes ASAP

asusevski commented 6 months ago

@merveenoyan @johko I know I'm fairly late, I only recently returned from spending time away for the holidays, I was hoping you could check now if all the changes are good to merge?

asusevski commented 6 months ago

@merveenoyan done! I thought i had removed it already for some reason. Sorry about that!!