From bca423ee435212f184c902e9316b3ef22e232d57 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 12 Jun 2019 14:15:28 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 436e157b..0df92b0e 100644 --- a/train.py +++ b/train.py @@ -207,7 +207,7 @@ def train( if opt.multi_scale: if (i + 1 + nb * epoch) % 10 == 0: #  adjust (67% - 150%) every 10 batches img_size = random.choice(range(img_size_min, img_size_max + 1)) * 32 - print('multi_scale img_size = %g' % img_size) + print('img_size = %g' % img_size) scale_factor = img_size / max(imgs.shape[-2:]) imgs = F.interpolate(imgs, scale_factor=scale_factor, mode='bilinear', align_corners=False)