Fix MultiCropTransform crop_max_scales check (#1454)

* Fix crop_max_scales check
lightly-ai · Dec 13, 2023 · 0802550 · 0802550
1 parent 7cf2970
commit 0802550
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lightly/transforms/multi_crop_transform.py b/lightly/transforms/multi_crop_transform.py
@@ -50,10 +50,10 @@ def __init__(
                 "Length of crop_sizes and crop_min_scales must be equal but are"
                 f" {len(crop_sizes)} and {len(crop_min_scales)}."
             )
-        if len(crop_sizes) != len(crop_min_scales):
+        if len(crop_sizes) != len(crop_max_scales):
             raise ValueError(
                 "Length of crop_sizes and crop_max_scales must be equal but are"
-                f" {len(crop_sizes)} and {len(crop_min_scales)}."
+                f" {len(crop_sizes)} and {len(crop_max_scales)}."
             )
 
         crop_transforms = []