diff --git a/src/mednet/libs/segmentation/config/models/driu.py b/src/mednet/libs/segmentation/config/models/driu.py
index 847a7cd38d62c219f2663152a3462f08168a07b0..867619ead324b2b655b23b37c5cda5f91570763c 100644
--- a/src/mednet/libs/segmentation/config/models/driu.py
+++ b/src/mednet/libs/segmentation/config/models/driu.py
@@ -39,5 +39,5 @@ model = DRIU(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/config/models/driu_bn.py b/src/mednet/libs/segmentation/config/models/driu_bn.py
index 8eddae7b609bf2303df8c7b242296e853d0a953b..52fa78f44741686e69ec31a58e97c7057d977bcd 100644
--- a/src/mednet/libs/segmentation/config/models/driu_bn.py
+++ b/src/mednet/libs/segmentation/config/models/driu_bn.py
@@ -39,5 +39,5 @@ model = DRIUBN(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/config/models/driu_od.py b/src/mednet/libs/segmentation/config/models/driu_od.py
index 0fd4b9f0f0c06e46f3056f0f413bb7c85ff1969f..5994dff0de6c146b7e48b4b426fc2304bb45ed8f 100644
--- a/src/mednet/libs/segmentation/config/models/driu_od.py
+++ b/src/mednet/libs/segmentation/config/models/driu_od.py
@@ -39,5 +39,5 @@ model = DRIUOD(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/config/models/driu_pix.py b/src/mednet/libs/segmentation/config/models/driu_pix.py
index e0365c885a2b3bcec22895eb8d23538af4146c97..f9ddb8487bf0864a6720f36a2fd9e79384e46c7b 100644
--- a/src/mednet/libs/segmentation/config/models/driu_pix.py
+++ b/src/mednet/libs/segmentation/config/models/driu_pix.py
@@ -39,5 +39,5 @@ model = DRIUPix(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/config/models/hed.py b/src/mednet/libs/segmentation/config/models/hed.py
index 0c88c9c262c03bd184c3da971af6e35e348e0b6d..d5d0f293e64df3ab30ac97b366cf8090d1a27c1f 100644
--- a/src/mednet/libs/segmentation/config/models/hed.py
+++ b/src/mednet/libs/segmentation/config/models/hed.py
@@ -15,7 +15,6 @@ final_lr = 0.1
 gamma = 1e-3
 eps = 1e-8
 amsbound = False
-crop_size = 544
 
 model = HED(
     loss_type=MultiSoftJaccardBCELogitsLoss,
@@ -31,5 +30,5 @@ model = HED(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=crop_size,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/config/models/lwnet.py b/src/mednet/libs/segmentation/config/models/lwnet.py
index b0428f4498f6da65ac6549e39c1fec21a7bbf265..262636ccc5ed4e575bd80e550090785549480eb5 100644
--- a/src/mednet/libs/segmentation/config/models/lwnet.py
+++ b/src/mednet/libs/segmentation/config/models/lwnet.py
@@ -22,5 +22,5 @@ model = LittleWNet(
     optimizer_type=Adam,
     optimizer_arguments=dict(lr=max_lr),
     augmentation_transforms=[],
-    crop_size=544,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/config/models/m2unet.py b/src/mednet/libs/segmentation/config/models/m2unet.py
index bb360709d279175b5b513bc3fcf30a986279b83f..ec47f328b4851b3fa56cc4021be965dba6b34ebd 100644
--- a/src/mednet/libs/segmentation/config/models/m2unet.py
+++ b/src/mednet/libs/segmentation/config/models/m2unet.py
@@ -28,7 +28,6 @@ final_lr = 0.1
 gamma = 1e-3
 eps = 1e-8
 amsbound = False
-crop_size = 544
 
 model = M2UNET(
     loss_type=SoftJaccardBCELogitsLoss,
@@ -44,5 +43,5 @@ model = M2UNET(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=crop_size,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/config/models/unet.py b/src/mednet/libs/segmentation/config/models/unet.py
index 10859a6775ebcdba73e471e6e731677125c49b8f..9919b12ec68ed4f236f3da628dc5fe5e89405b74 100644
--- a/src/mednet/libs/segmentation/config/models/unet.py
+++ b/src/mednet/libs/segmentation/config/models/unet.py
@@ -41,5 +41,5 @@ model = Unet(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )
diff --git a/src/mednet/libs/segmentation/models/driu.py b/src/mednet/libs/segmentation/models/driu.py
index 6afab13371324d32816537469e47a691889ac1c8..fc655cecdba8792523bcef79f8ba09dcc1dc67ed 100644
--- a/src/mednet/libs/segmentation/models/driu.py
+++ b/src/mednet/libs/segmentation/models/driu.py
@@ -110,7 +110,7 @@ class DRIU(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 1024,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,
diff --git a/src/mednet/libs/segmentation/models/driu_bn.py b/src/mednet/libs/segmentation/models/driu_bn.py
index 3f8d89937d3ce1602d6c75a4057768c620427dc2..96b019907279eeaee79b5f52c21d38803a94a3bc 100644
--- a/src/mednet/libs/segmentation/models/driu_bn.py
+++ b/src/mednet/libs/segmentation/models/driu_bn.py
@@ -113,7 +113,7 @@ class DRIUBN(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,
diff --git a/src/mednet/libs/segmentation/models/driu_od.py b/src/mednet/libs/segmentation/models/driu_od.py
index f2deb065c1461d0f25f55711f7621a059a327ab9..9272d3919238b4d3c4a710822cb6e564fe590b09 100644
--- a/src/mednet/libs/segmentation/models/driu_od.py
+++ b/src/mednet/libs/segmentation/models/driu_od.py
@@ -95,7 +95,7 @@ class DRIUOD(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,
diff --git a/src/mednet/libs/segmentation/models/driu_pix.py b/src/mednet/libs/segmentation/models/driu_pix.py
index 08f721d0e2a21a512782c3a90f1299b8c68d5be5..e84d45dbc8f9699ea37f67a70e779551e6b81f0d 100644
--- a/src/mednet/libs/segmentation/models/driu_pix.py
+++ b/src/mednet/libs/segmentation/models/driu_pix.py
@@ -99,7 +99,7 @@ class DRIUPix(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,
diff --git a/src/mednet/libs/segmentation/models/hed.py b/src/mednet/libs/segmentation/models/hed.py
index 7ac25fb45980b8997b76161a317c59abbba197fb..c20286fc52861e9d0aceccb4f90a92214584aaad 100644
--- a/src/mednet/libs/segmentation/models/hed.py
+++ b/src/mednet/libs/segmentation/models/hed.py
@@ -113,7 +113,7 @@ class HED(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,
diff --git a/src/mednet/libs/segmentation/models/lwnet.py b/src/mednet/libs/segmentation/models/lwnet.py
index b9519e734331788443d0205dde13844713eb9d2c..041b8ce3403171225618f4fe8c0b5ef1c4b68782 100644
--- a/src/mednet/libs/segmentation/models/lwnet.py
+++ b/src/mednet/libs/segmentation/models/lwnet.py
@@ -310,7 +310,7 @@ class LittleWNet(Model):
         optimizer_arguments: dict[str, typing.Any] = {},
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,
diff --git a/src/mednet/libs/segmentation/models/m2unet.py b/src/mednet/libs/segmentation/models/m2unet.py
index 76042ba2b3d35da9058b128edd9530e49c3f010a..22ee9ee6ab0ab400638ec87a0ccc05155e26efd9 100644
--- a/src/mednet/libs/segmentation/models/m2unet.py
+++ b/src/mednet/libs/segmentation/models/m2unet.py
@@ -161,7 +161,7 @@ class M2UNET(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,
diff --git a/src/mednet/libs/segmentation/models/unet.py b/src/mednet/libs/segmentation/models/unet.py
index cc478a3c5f7d19709a8f849db1b37db520ac8456..f050e60c9b39267845a001dea337316b8d16816d 100644
--- a/src/mednet/libs/segmentation/models/unet.py
+++ b/src/mednet/libs/segmentation/models/unet.py
@@ -102,7 +102,7 @@ class Unet(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,