updates

2019-12-24 14:11:31 -08:00 · 2019-12-24 14:11:31 -08:00 · cdc382e313
parent 2ee0d0c714
commit cdc382e313
1 changed files with 1 additions and 1 deletions
--- 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(opt.device, batch_size=batch_size)
-        verbose = True
+        verbose = opt.task == 'test'

        # Remove previous
        for f in glob.glob('test_batch*.jpg'):