8 Commits

Author	SHA1	Message	Date
regisss	0cdcd7a2b3	Remove task arg in load_dataset in image-classification example (#28408) ... * Remove `task` arg in `load_dataset` in image-classification example * Manage case where "train" is not in dataset * Add new args to manage image and label column names * Similar to audio-classification example * Fix README * Update tests	2024-01-16 08:04:08 +01:00
wei zhao	7d5ce6802e	Fix typo in image-classification/README.md (#19424) ... Fix link typo of the following content. PyTorch version, Trainer PyTorch version, no Trainer	2022-10-10 09:16:58 -04:00
regisss	28a0811652	Improve mismatched sizes management when loading a pretrained model (#17257) ... - Add --ignore_mismatched_sizes argument to classification examples - Expand the error message when loading a model whose head dimensions are different from expected dimensions	2022-05-17 17:58:14 +02:00
Zachary Mueller	d719bcd46a	Fix all docs for accelerate install directions (#17145)	2022-05-09 15:45:18 -04:00
NielsRogge	b96e82c80a	Add image classification script, no trainer (#16727) ... * Add first draft * Improve README and run fixup * Make script aligned with other scripts, improve README * Improve script and add test * Remove print statement * Apply suggestions from code review * Add num_labels to make test pass * Improve README	2022-04-19 16:32:08 +02:00
NielsRogge	78f346c2b5	Update README.md (#16797)	2022-04-15 14:10:16 +02:00
NielsRogge	048443db86	Improve image classification example (#16585) ... * 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>	2022-04-14 18:10:52 +02:00
Nathan Raw	76c4d8bf26	✨ Add PyTorch image classification example (#13134) ... * ✨ add pytorch image classification example * 🔥 remove utils.py * 💄 fix flake8 style issues * 🔥 remove unnecessary line * ✨ limit dataset sizes * 📌 update reqs * 🎨 restructure - use datasets lib * 🎨 import transforms directly * 📝 add comments * 💄 style * 🔥 remove flag * 📌 update requirement warning * 📝 add vision README.md * 📝 update README.md * 📝 update README.md * 🎨 add image-classification tag to model card * 🚚 rename vision ➡️ image-classification * 📝 update image-classification README.md	2021-09-02 13:29:42 -06:00