From 4a5d97bcf145bbb768aeec28ed015a079639b778 Mon Sep 17 00:00:00 2001 From: Prathik Rao Date: Sun, 25 Feb 2024 04:19:12 -0800 Subject: [PATCH] Image Classification Example Bug Fix (#1718) --- .../training/image-classification/run_image_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/onnxruntime/training/image-classification/run_image_classification.py b/examples/onnxruntime/training/image-classification/run_image_classification.py index bc5482e23e..c5d5aabe27 100644 --- 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"""