diff --git a/templates/repo/modelmanage/showinfo.tmpl b/templates/repo/modelmanage/showinfo.tmpl index bb4807c07..0a57a5f44 100644 --- a/templates/repo/modelmanage/showinfo.tmpl +++ b/templates/repo/modelmanage/showinfo.tmpl @@ -232,13 +232,25 @@ function loadInfo(){ renderInfo(initObj,initModelAcc,id) loadModelFile(data[0].ID,data[0].Version,'','','init') }) +} + function getEngineName(model){ + if(model.Engine == 0){ + return "Pytorch"; + }else if(model.Engine == 1 || model.Engine == 121){ + return "TensorFlow"; + }else if(model.Engine == 2 || model.Engine == 122){ + return "MindSpore"; + }else{ + return "Other" + } } function transObj(data){ let {ID,Name,Version,Label,Size,Description,CreatedUnix,Accuracy,CodeBranch,CodeCommitID,TrainTaskInfo} = data[0] let modelAcc = JSON.parse(Accuracy) TrainTaskInfo = JSON.parse(TrainTaskInfo) // Parameters = JSON.parse(Parameters) - let {Parameters,EngineName} = TrainTaskInfo + let {Parameters} = TrainTaskInfo + let EngineName = getEngineName(data[0]) Parameters = JSON.parse(Parameters) Parameters = Parameters.parameter.length === 0 ? '--':Parameters.parameter let size = tranSize(Size)