Skip to content

Conversation

NielsRogge
Copy link
Contributor

@NielsRogge NielsRogge commented Apr 4, 2022

What does this PR do?

This PR improves the image classification example script, by:

  • replacing the nateraw/imagefolder hack with ImageFolder
  • improving the README, explaining how to easily train on custom data.

@NielsRogge NielsRogge force-pushed the add_imagefolder_mention branch from c5c28dd to 52bde39 Compare April 4, 2022 14:50
@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Apr 4, 2022

The documentation is not available anymore as the PR was closed or merged.

@NielsRogge NielsRogge changed the title [WIP] Improve image classification example README [WIP] Improve image classification example Apr 4, 2022
@NielsRogge NielsRogge requested review from nateraw and sgugger April 12, 2022 13:36
@NielsRogge NielsRogge changed the title [WIP] Improve image classification example Improve image classification example Apr 12, 2022
Copy link
Collaborator

@sgugger sgugger left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Very cool, thanks a lot for making this script better!

@NielsRogge NielsRogge merged commit 048443d into huggingface:main Apr 14, 2022
elusenji pushed a commit to elusenji/transformers that referenced this pull request Jun 12, 2022
* Improve README

* Make dataset_name argument optional

* Improve local data

* Fix bug

* Improve README some more

* Apply suggestions from code review

* Improve README

Co-authored-by: Niels Rogge <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants