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

updates

This commit is contained in:
Glenn Jocher 2019-07-21 21:28:38 +02:00
parent aedf45c713
commit e97cc07715
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -21,7 +21,7 @@ def test(cfg,
# Initialize/load model and set device # Initialize/load model and set device
if model is None: if model is None:
device = torch_utils.select_device() device = torch_utils.select_device()
verbose = False verbose = True
# Initialize model # Initialize model
model = Darknet(cfg, img_size).to(device) model = Darknet(cfg, img_size).to(device)
@ -36,7 +36,7 @@ def test(cfg,
model = nn.DataParallel(model) model = nn.DataParallel(model)
else: else:
device = next(model.parameters()).device # get model device device = next(model.parameters()).device # get model device
verbose = True verbose = False
# Configure run # Configure run
data = parse_data_cfg(data) data = parse_data_cfg(data)