examples/pytorch/image-classification/run_image_classification.py · 048443db863214aef9c8341517b427edced63c81 · chenpangpang / transformers

Improve image classification example (#16585) · 048443db

NielsRogge authored Apr 14, 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 <nielsrogge@Nielss-MacBook-Pro.local>

048443db

run_image_classification.py 14.1 KB

Replace run_image_classification.py