From caeba13b843a16a26e33699989c0895ce4a583be Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 2 Sep 2018 14:01:31 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index f404e3d7..6fff74c5 100644 --- a/train.py +++ b/train.py @@ -95,7 +95,7 @@ def main(opt): # Multi-Scale Training # img_size = random.choice(range(10, 20)) * 32 - # dataloader = ListDataset(train_path, batch_size=opt.batch_size, img_size=img_size) + # dataloader = load_images_and_labels(train_path, batch_size=opt.batch_size, img_size=img_size, augment=True) # print('Running this epoch with image size %g' % img_size) # Update scheduler