fix (#972)

Deci-AI · May 10, 2023 · cb3ea56 · cb3ea56
1 parent e87400f
commit cb3ea56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/super_gradients/training/transforms/transforms.py b/src/super_gradients/training/transforms/transforms.py
@@ -850,7 +850,7 @@ def __call__(self, sample: dict) -> dict:
         return sample
 
     def get_equivalent_preprocessing(self) -> List[Dict]:
-        return [{Processings.DetectionRescale: {"output_shape": self.output_size}}]
+        return [{Processings.DetectionRescale: {"output_shape": self.output_shape}}]
 
 
 @register_transform(Transforms.DetectionRandomRotate90)