diff --git a/train.py b/train.py
index 0e87226..5395af7 100644
--- a/train.py
+++ b/train.py
@@ -84,8 +84,8 @@ def train_model():
     BATCH_SIZE = 32
     LEARNING_RATE = 0.001
     # For the custom crop transform. User can adjust these.
-    TRIANGLE_CROP_WIDTH = 400
-    TRIANGLE_CROP_HEIGHT = 400
+    TRIANGLE_CROP_WIDTH = 556
+    TRIANGLE_CROP_HEIGHT = 1184
     RESIZE_DIM = 64 # Resize cropped image to this dimension (square)
 
     # --- Data Preparation ---