Add TF image classification example script (#19956)
* TF image classification script * Update requirements * Fix up * Add tests * Update test fetcher Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * Fix directory path * Adding `zero-shot-object-detection` pipeline doctest. (#20274) * Adding `zero-shot-object-detection` pipeline doctest. * Remove nested_simplify. * Add generate kwargs to `AutomaticSpeechRecognitionPipeline` (#20952) * Add generate kwargs to AutomaticSpeechRecognitionPipeline * Add test for generation kwargs * Trigger CI * Data collator returns np * Update feature extractor -> image processor * Bug fixes - updates to reflect changes in API * Update flags to match PT & run faster * Update instructions - Maria's comment * Update examples/tensorflow/image-classification/README.md * Remove slow decorator --------- Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com> Co-authored-by: bofeng huang <bofenghuang7@gmail.com> Co-authored-by: Sylvain Gugger <Sylvain.gugger@gmail.com>
This commit is contained in:
@@ -565,6 +565,8 @@ def infer_tests_to_run(output_file, diff_with_last_commit=False, filters=None, j
|
||||
elif f.startswith("examples/pytorch"):
|
||||
test_files_to_run.append("examples/pytorch/test_pytorch_examples.py")
|
||||
test_files_to_run.append("examples/pytorch/test_accelerate_examples.py")
|
||||
elif f.startswith("examples/tensorflow"):
|
||||
test_files_to_run.append("examples/tensorflow/test_tensorflow_examples.py")
|
||||
elif f.startswith("examples/flax"):
|
||||
test_files_to_run.append("examples/flax/test_flax_examples.py")
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user