Skip to content

Commit e7abc32

Browse files
committed
Improve bulk_runner for 'all resolution' mode
1 parent 40d55ab commit e7abc32

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

bulk_runner.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,13 +113,14 @@ def main():
113113
model_cfgs.append((n, None))
114114
elif args.model_list == 'all_res':
115115
model_names = list_models()
116-
model_names += [n.split('.')[0] for n in list_models(pretrained=True)]
116+
model_names += list_models(pretrained=True)
117117
model_cfgs = set()
118118
for n in model_names:
119119
pt_cfg = get_pretrained_cfg(n)
120120
if pt_cfg is None:
121121
print(f'Model {n} is missing pretrained cfg, skipping.')
122122
continue
123+
n = n.split('.')[0]
123124
model_cfgs.add((n, pt_cfg.input_size[-1]))
124125
if pt_cfg.test_input_size is not None:
125126
model_cfgs.add((n, pt_cfg.test_input_size[-1]))

0 commit comments

Comments
 (0)