From e97cc07715aad8627e5caf078b06478e1e3c4a04 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 21 Jul 2019 21:28:38 +0200 Subject: [PATCH] updates --- test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index 21333788..c16790e0 100644 --- a/test.py +++ b/test.py @@ -21,7 +21,7 @@ def test(cfg, # Initialize/load model and set device if model is None: device = torch_utils.select_device() - verbose = False + verbose = True # Initialize model model = Darknet(cfg, img_size).to(device) @@ -36,7 +36,7 @@ def test(cfg, model = nn.DataParallel(model) else: device = next(model.parameters()).device # get model device - verbose = True + verbose = False # Configure run data = parse_data_cfg(data)