From 2a8c4dc63bfd9431e66f256b27ab7dd2fccca786 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 7 Feb 2021 21:35:50 -0800 Subject: [PATCH] Add validation script update for using test_input_size in model default_cfgs --- validate.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/validate.py b/validate.py index 5b5f98cf..8ad9cb1f 100755 --- a/validate.py +++ b/validate.py @@ -147,8 +147,10 @@ def validate(args): param_count = sum([m.numel() for m in model.parameters()]) _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) + data_config = resolve_data_config(vars(args), model=model, use_test_size=True) + test_time_pool = False + if not args.no_test_pool: + model, test_time_pool = apply_test_time_pool(model, data_config, use_test_size=True) if args.torchscript: torch.jit.optimized_execution(True)