From 837a4dddb8a484a249c5d41478600ff5cb8e5841 Mon Sep 17 00:00:00 2001 From: Yuta Hayashibe Date: Thu, 26 Oct 2023 13:34:36 +0900 Subject: [PATCH] Added assertions --- library/train_util.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/library/train_util.py b/library/train_util.py index 51610e70..a38e7978 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -148,6 +148,13 @@ class ImageInfo: class BucketManager: def __init__(self, no_upscale, max_reso, min_size, max_size, reso_steps) -> None: + if max_size is not None: + if max_reso is not None: + assert max_size >= max_reso[0], "the max_size should be larger than the width of max_reso" + assert max_size >= max_reso[1], "the max_size should be larger than the height of max_reso" + if min_size is not None: + assert max_size >= min_size, "the max_size should be larger than the min_size" + self.no_upscale = no_upscale if max_reso is None: self.max_reso = None