From c670e3fc37f132a07b2f1fe9dec6e7f4f3875dd5 Mon Sep 17 00:00:00 2001
From: Joshua David <jshuadvd@icloud.com>
Date: Fri, 5 Jul 2024 22:11:59 -0700
Subject: [PATCH] Swap print statement for logger

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index de120da..ab5870b 100644
--- a/train.py
+++ b/train.py
@@ -266,7 +266,7 @@ def train(
         else:
             patience += 1
             if patience >= max_patience:
-                print("Early stopping triggered")
+                logger.info("Early stopping triggered")
                 break