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

find model with upper name.

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