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 = []