Image Classification Example Bug Fix (#1718)

huggingface · Feb 25, 2024 · 4a5d97b · 4a5d97b
1 parent b5f4de3
commit 4a5d97b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/onnxruntime/training/image-classification/run_image_classification.py b/examples/onnxruntime/training/image-classification/run_image_classification.py
@@ -48,7 +48,7 @@
 from transformers.utils import check_min_version, send_example_telemetry
 from transformers.utils.versions import require_version
 
-from optimum import ORTTrainer, ORTTrainingArguments
+from optimum.onnxruntime import ORTTrainer, ORTTrainingArguments
 
 
 """ Fine-tuning a 🤗 Transformers model for image classification"""