apply device to validators (#1216)

deepchecks/vision/utils/validation.py

@@ -19,6 +19,7 @@
 from deepchecks.core.errors import ValidationError
 from deepchecks.utils.ipython import is_headless, is_notebook
 from deepchecks.utils.strings import create_new_file_name
+from deepchecks.vision.batch_wrapper import apply_to_tensor
 from deepchecks.vision.vision_data import TaskType
 from deepchecks.vision.utils.image_functions import numpy_to_image_figure, label_bbox_add_to_figure
 from deepchecks.vision.vision_data import VisionData
@@ -71,7 +72,7 @@ def validate_extractors(dataset: VisionData, model, device=None, image_save_loca
         if the value is True.
     """
     print('Deepchecks will try to validate the extractors given...')
-    batch = next(iter(dataset.data_loader))
+    batch = apply_to_tensor(next(iter(dataset.data_loader)), lambda it: it.to(device))
     images = None
     labels = None
     predictions = None