From 8aff2583a16f277b3a0477faa374a9304eeacf52 Mon Sep 17 00:00:00 2001
From: jp-ga <97850904+jp-ga@users.noreply.github.com>
Date: Mon, 29 Jun 2026 15:23:33 -0700
Subject: [PATCH] add epsilon to normalization

---
 gpsr/losses.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpsr/losses.py b/gpsr/losses.py
index e3ce0c9..d2b24c6 100644
--- a/gpsr/losses.py
+++ b/gpsr/losses.py
@@ -21,7 +21,7 @@ def normalize_images(images):
     """
 
     sums = images.sum(dim=(-1, -2), keepdim=True)
-    return images / sums
+    return images / sums.clamp_min(torch.finfo(images.dtype).eps)
 
 
 def kl_div(target, pred):