diff --git a/bulk_runner.py b/bulk_runner.py index 5f5db640..37541557 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]))