We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 40d55ab commit e7abc32Copy full SHA for e7abc32
bulk_runner.py
@@ -113,13 +113,14 @@ def main():
113
model_cfgs.append((n, None))
114
elif args.model_list == 'all_res':
115
model_names = list_models()
116
- model_names += [n.split('.')[0] for n in list_models(pretrained=True)]
+ model_names += list_models(pretrained=True)
117
model_cfgs = set()
118
for n in model_names:
119
pt_cfg = get_pretrained_cfg(n)
120
if pt_cfg is None:
121
print(f'Model {n} is missing pretrained cfg, skipping.')
122
continue
123
+ n = n.split('.')[0]
124
model_cfgs.add((n, pt_cfg.input_size[-1]))
125
if pt_cfg.test_input_size is not None:
126
model_cfgs.add((n, pt_cfg.test_input_size[-1]))
0 commit comments