提交 29f0b83a 编写于 作者: D dengkaipeng

find model with upper name.

上级 7e541e91
......@@ -169,7 +169,7 @@ class ModelZoo(object):
def get(self, name, cfg, mode='train'):
for k, v in self.model_zoo.items():
if k == name:
if k.upper() == name.upper():
return v(name, cfg, mode)
raise ModelNotFoundError(name, self.model_zoo.keys())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册