From e58f7bbdf0e5f04e34e1ec69cfbef03003c58776 Mon Sep 17 00:00:00 2001 From: Peter Lin Date: Wed, 10 Nov 2021 09:54:49 -0800 Subject: [PATCH] Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index cf24bb9..462bd1f 100644 --- a/train.py +++ b/train.py @@ -473,7 +473,7 @@ class Trainer: def random_crop(self, *imgs): h, w = imgs[0].shape[-2:] w = random.choice(range(w // 2, w)) - h = random.choice(range(w // 2, h)) + h = random.choice(range(h // 2, h)) results = [] for img in imgs: B, T = img.shape[:2]