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)