fix: update triangle crop dimensions

2025-07-31 16:06:27 -06:00
parent 4d6eb70728
commit 62743006c8
1 changed files with 2 additions and 2 deletions
--- 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 ---