Skip to content

Commit 8ffdc59

Browse files
committed
test_time_pool would be set to a non-False value even if test-time pooling is not available
1 parent 4be5b51 commit 8ffdc59

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: validate.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def validate(args):
139139
_logger.info('Model %s created, param count: %d' % (args.model, param_count))
140140

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

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

0 commit comments

Comments
 (0)