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
This commit is contained in:
regisss
2024-01-16 08:04:08 +01:00
committed by GitHub
parent edb170238f
commit 0cdcd7a2b3
5 changed files with 71 additions and 18 deletions

View File

@@ -398,6 +398,7 @@ class ExamplesTests(TestCasePlus):
--max_steps 10
--train_val_split 0.1
--seed 42
--label_column_name labels
""".split()
if is_torch_fp16_available_on_device(torch_device):