diff --git a/scripts/benchmarks/compare_with_anomalib.py b/scripts/benchmarks/compare_with_anomalib.py index d1fbaa9..ed0795a 100644 --- a/scripts/benchmarks/compare_with_anomalib.py +++ b/scripts/benchmarks/compare_with_anomalib.py @@ -810,6 +810,7 @@ def _anomalib_preprocess(model, images): def _anomalib_raw(model, images): """Torch model on already pre-processed images, skipping Anomalib's post-processor (min-max normalisation + clamping), like AnomaVision.""" + model.model = model.model.to(images.device) output = model.model(images) if not getattr(model, "_debug_features_printed", False):