From 62743006c8ec9e7e125ffe048d38e55aa6beebc2 Mon Sep 17 00:00:00 2001
From: Tanner Collin <git@tannercollin.com>
Date: Thu, 31 Jul 2025 16:06:27 -0600
Subject: [PATCH] fix: update triangle crop dimensions

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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 ---