From 29f0b83afadb59db4859b3b99eb50370967ba789 Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Mon, 8 Apr 2019 15:00:53 +0800 Subject: [PATCH] find model with upper name. --- PaddleCV/video/models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PaddleCV/video/models/model.py b/PaddleCV/video/models/model.py index 41b5b655..5bc354be 100644 --- a/PaddleCV/video/models/model.py +++ b/PaddleCV/video/models/model.py @@ -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()) -- GitLab