Merge a5ed288c09 into 1dae34b0af

Compute aspect ratios in log-space
2026-04-10 15:00:23 +00:00 · 2026-04-01 15:38:45 +00:00 · 2024-10-30 11:57:48 -04:00
1 changed files with 1 additions and 1 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -292,7 +292,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]
Author	SHA1	Message	Date
catboxanon	aa9b38ffea	Merge `a5ed288c09` into `1dae34b0af`	2026-04-01 15:38:45 +00:00
catboxanon	a5ed288c09	Compute aspect ratios in log-space	2024-10-30 11:57:48 -04:00