diff --git a/examples/image-classification/requirements.txt b/examples/image-classification/requirements.txt index fc17d6b79f..87694059fe 100644 --- a/examples/image-classification/requirements.txt +++ b/examples/image-classification/requirements.txt @@ -1,5 +1,5 @@ torch>=1.5.0 torchvision>=0.6.0 -datasets>=2.4.0 +datasets>=2.14.0 evaluate scikit-learn diff --git a/examples/image-classification/run_image_classification.py b/examples/image-classification/run_image_classification.py index 080970c344..82b2e1ba3f 100644 --- a/examples/image-classification/run_image_classification.py +++ b/examples/image-classification/run_image_classification.py @@ -67,7 +67,7 @@ def check_optimum_habana_min_version(*a, **b): check_min_version("4.34.0") check_optimum_habana_min_version("1.8.1") -require_version("datasets>=1.8.0", "To fix: pip install -r examples/pytorch/image-classification/requirements.txt") +require_version("datasets>=2.14.0", "To fix: pip install -r examples/pytorch/image-classification/requirements.txt") MODEL_CONFIG_CLASSES = list(MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING.keys()) MODEL_TYPES = tuple(conf.model_type for conf in MODEL_CONFIG_CLASSES)