From d0be1214bf37865ab42a89e384ddb00e683833c3 Mon Sep 17 00:00:00 2001 From: jp-ga <97850904+jp-ga@users.noreply.github.com> Date: Wed, 1 Jul 2026 00:07:34 -0700 Subject: [PATCH] change eps to tiny --- gpsr/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gpsr/losses.py b/gpsr/losses.py index d2b24c6..a050d7c 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.clamp_min(torch.finfo(images.dtype).eps) + return images / sums.clamp_min(torch.finfo(images.dtype).tiny) def kl_div(target, pred):