From c94019f159f6ac8148288a510f72ea9b33ce78b9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 17 May 2020 14:31:14 -0700 Subject: [PATCH] iglob bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index d3bd51fd..d573890c 100644 --- a/train.py +++ b/train.py @@ -398,7 +398,7 @@ if __name__ == '__main__': opt.weights = last if opt.resume else opt.weights check_git_status() opt.cfg = list(glob.iglob('./**/' + opt.cfg, recursive=True))[0] # find file - opt.data = list(glob.iglob('./**/' + opt.data, recursive=True))[0] # find file + # opt.data = list(glob.iglob('./**/' + opt.data, recursive=True))[0] # find file print(opt) opt.img_size.extend([opt.img_size[-1]] * (3 - len(opt.img_size))) # extend to 3 sizes (min, max, test) device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size)