From ddbc07c09cfa077a20eeb983e863dd5c7057ed09 Mon Sep 17 00:00:00 2001 From: zouap Date: Tue, 12 Jul 2022 16:06:49 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=A0=B7=E5=BC=8F=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- templates/repo/modelmanage/showinfo.tmpl | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/templates/repo/modelmanage/showinfo.tmpl b/templates/repo/modelmanage/showinfo.tmpl index 846134ece..a4577fe20 100644 --- a/templates/repo/modelmanage/showinfo.tmpl +++ b/templates/repo/modelmanage/showinfo.tmpl @@ -237,9 +237,18 @@ function loadInfo(){ return "PyTorch"; }else if(model.Engine == 1 || model.Engine == 121){ return "TensorFlow"; - }else if(model.Engine == 2 || model.Engine == 122){ + }else if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35){ return "MindSpore"; - }else{ + }else if(model.Engine == 3){ + return "Other"; + }else if(model.Engine == 4){ + return "PaddlePaddle"; + }else if(model.Engine == 5){ + return "OneFlow"; + }else if(model.Engine == 6){ + return "MXNet"; + } + else{ return "Other" } }