Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add _in22k to bulk validate filter

Browse Source
pull/571/head
Ross Wightman 4 years ago
parent af647c10b3
commit a04427d8ce
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
validate.py
Unescape View File

@ -289,7 +289,7 @@ def main():
if args.model == 'all': if args.model == 'all':
# validate all models in a list of names with pretrained checkpoints # validate all models in a list of names with pretrained checkpoints
args.pretrained = True args.pretrained = True
model_names = list_models(pretrained=True, exclude_filters=['*in21k']) model_names = list_models(pretrained=True, exclude_filters=['*_in21k', '*_in22k'])
model_cfgs = [(n, '') for n in model_names] model_cfgs = [(n, '') for n in model_names]
elif not is_model(args.model): elif not is_model(args.model):
# model name doesn't exist, try as wildcard filter # model name doesn't exist, try as wildcard filter

Write Preview
Loading…
Cancel
Save