test_time_pool would be set to a non-False value even if test-time pooling is not available

4 years ago · 8ffdc5910a
parent 4be5b51e0a
commit 8ffdc5910a
1 changed files with 1 additions and 1 deletions
--- a/validate.py
+++ b/validate.py
@ -139,7 +139,7 @@ def validate(args):
    _logger.info('Model %s created, param count: %d' % (args.model, param_count))

    data_config = resolve_data_config(vars(args), model=model)
-    model, test_time_pool = model, False if args.no_test_pool else apply_test_time_pool(model, data_config)
+    model, test_time_pool = (model, False) if args.no_test_pool else apply_test_time_pool(model, data_config)

    if args.torchscript:
        torch.jit.optimized_execution(True)