From a5ed288c09f628edf55cce965ffd74a318fcdfb3 Mon Sep 17 00:00:00 2001
From: catboxanon <122327233+catboxanon@users.noreply.github.com>
Date: Wed, 30 Oct 2024 11:57:48 -0400
Subject: [PATCH] Compute aspect ratios in log-space

---
 library/train_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/library/train_util.py b/library/train_util.py
index 100ef475..878b1cd2 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -245,7 +245,7 @@ class BucketManager:
             if reso in self.predefined_resos_set:
                 pass
             else:
-                ar_errors = self.predefined_aspect_ratios - aspect_ratio
+                ar_errors = np.log(self.predefined_aspect_ratios) - np.log(aspect_ratio)
                 predefined_bucket_id = np.abs(ar_errors).argmin()  # 当該解像度以外でaspect ratio errorが最も少ないもの
                 reso = self.predefined_resos[predefined_bucket_id]