zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

prevent testloader caching on --notest

This commit is contained in:
Glenn Jocher 2020-09-11 16:59:13 -07:00
parent 35dc935504
commit a62a45b2dd
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
train.py
View File

@ -165,8 +165,8 @@ def train(hyp, opt, device, tb_writer=None):
# Trainloader # Trainloader
dataloader, dataset = create_dataloader(train_path, imgsz, batch_size, gs, opt, dataloader, dataset = create_dataloader(train_path, imgsz, batch_size, gs, opt,
hyp=hyp, augment=True, cache=opt.cache_images, rect=opt.rect, rank=rank, hyp=hyp, augment=True, cache=opt.cache_images, rect=opt.rect,
world_size=opt.world_size, workers=opt.workers) rank=rank, world_size=opt.world_size, workers=opt.workers)
mlc = np.concatenate(dataset.labels, 0)[:, 0].max() # max label class mlc = np.concatenate(dataset.labels, 0)[:, 0].max() # max label class
nb = len(dataloader) # number of batches nb = len(dataloader) # number of batches
assert mlc < nc, 'Label class %g exceeds nc=%g in %s. Possible class labels are 0-%g' % (mlc, nc, opt.data, nc - 1) assert mlc < nc, 'Label class %g exceeds nc=%g in %s. Possible class labels are 0-%g' % (mlc, nc, opt.data, nc - 1)
@ -175,8 +175,8 @@ def train(hyp, opt, device, tb_writer=None):
if rank in [-1, 0]: if rank in [-1, 0]:
ema.updates = start_epoch * nb // accumulate # set EMA updates ema.updates = start_epoch * nb // accumulate # set EMA updates
testloader = create_dataloader(test_path, imgsz_test, total_batch_size, gs, opt, testloader = create_dataloader(test_path, imgsz_test, total_batch_size, gs, opt,
hyp=hyp, augment=False, cache=opt.cache_images, rect=True, rank=-1, hyp=hyp, augment=False, cache=opt.cache_images and not opt.notest, rect=True,
world_size=opt.world_size, workers=opt.workers)[0] # testloader rank=-1, world_size=opt.world_size, workers=opt.workers)[0] # testloader
if not opt.resume: if not opt.resume:
labels = np.concatenate(dataset.labels, 0) labels = np.concatenate(dataset.labels, 0)