Skip to content
Snippets Groups Projects
Commit 5bcb9fce authored by andong's avatar andong
Browse files

correctly print model_names

parent c0b5dae3
No related branches found
No related tags found
No related merge requests found
......@@ -36,7 +36,7 @@ def create_model(arch, pretrained='', delete_fc=False, *args, **kwargs):
# creating model
if arch not in globals():
raise NameError("unknown model architecture '%s'\nSelect one in %s" % (
arch, ','.join(list_archs())))
arch, ','.join(model_names)))
model = globals()[arch](*args, **kwargs)
model.preprocess = dict(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment