See if this fixes

Lightning-Universe · Jun 22, 2022 · 7c55240 · 7c55240
1 parent bb78340
commit 7c55240
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/task/vision/test_image_classification.py b/tests/task/vision/test_image_classification.py
@@ -17,7 +17,7 @@
 def test_smoke_train(hf_cache_path):
     feature_extractor = AutoFeatureExtractor.from_pretrained(pretrained_model_name_or_path="nateraw/tiny-vit-random")
     dm = ImageClassificationDataModule(
-        cfg=ImageClassificationDataConfig(batch_size=2, dataset_name="beans"),
+        cfg=ImageClassificationDataConfig(batch_size=1, dataset_name="beans"),
         feature_extractor=feature_extractor,
     )
     model = ImageClassificationTransformer(