From e7abc320f53a170f9f73e4ec1584e73a82ff09cf Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 23 Nov 2023 12:44:07 -0800 Subject: [PATCH] Improve bulk_runner for 'all resolution' mode --- bulk_runner.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/bulk_runner.py b/bulk_runner.py index 5f5db6407e..37541557ac 100755 --- a/bulk_runner.py +++ b/bulk_runner.py @@ -113,13 +113,14 @@ def main(): model_cfgs.append((n, None)) elif args.model_list == 'all_res': model_names = list_models() - model_names += [n.split('.')[0] for n in list_models(pretrained=True)] + model_names += list_models(pretrained=True) model_cfgs = set() for n in model_names: pt_cfg = get_pretrained_cfg(n) if pt_cfg is None: print(f'Model {n} is missing pretrained cfg, skipping.') continue + n = n.split('.')[0] model_cfgs.add((n, pt_cfg.input_size[-1])) if pt_cfg.test_input_size is not None: model_cfgs.add((n, pt_cfg.test_input_size[-1]))