apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher 2019-12-14 15:40:01 -08:00
parent eb70e4b751
commit 8c13717f48
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -208,7 +208,8 @@ 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, opt.img_size, batch_size,
hyp=hyp,
rect=True, rect=True,
cache_labels=True, cache_labels=True,
cache_images=opt.cache_images), cache_images=opt.cache_images),