Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-07 22:55:26 -08:00
parent 1f943e886f
commit b81beb0f5f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -210,6 +210,7 @@ def train():
# Test Dataloader # Test Dataloader
if not opt.prebias: if not opt.prebias:
testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, img_size, batch_size, hyp=hyp, testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, img_size, batch_size, hyp=hyp,
rect=True,
cache_labels=True, cache_labels=True,
cache_images=opt.cache_images), cache_images=opt.cache_images),
batch_size=batch_size, batch_size=batch_size,