diff --git a/templates/repo/modelarts/trainjob/show.tmpl b/templates/repo/modelarts/trainjob/show.tmpl index 1cdb595dd..e95a1233d 100755 --- a/templates/repo/modelarts/trainjob/show.tmpl +++ b/templates/repo/modelarts/trainjob/show.tmpl @@ -892,11 +892,11 @@ $('#JobName').val(obj.DisplayJobName).addClass('model_disabled') $('input[name="JobId"]').val(obj.JobID) $('input[name="VersionName"]').val(obj.VersionName).addClass('model_disabled') - if(obj.EngineID ==122 || obj.EngineID ==35 || obj.EngineID ==-1){ + if(obj.EngineID ==122 || obj.EngineID ==35 || obj.EngineID ==-1 || obj.EngineID ==37){ $('input[name="Engine_name"]').val("MindSpore").addClass('model_disabled'); $('input[name="Engine"]').val(2); } - if(obj.EngineID ==121){ + if(obj.EngineID ==121 || obj.EngineID ==38){ $('input[name="Engine_name"]').val("TensorFlow").addClass('model_disabled'); $('input[name="Engine"]').val(1); } diff --git a/templates/repo/modelmanage/convertIndex.tmpl b/templates/repo/modelmanage/convertIndex.tmpl index 3c4ac7f69..92eefca2e 100644 --- a/templates/repo/modelmanage/convertIndex.tmpl +++ b/templates/repo/modelmanage/convertIndex.tmpl @@ -581,10 +581,10 @@ if(versionList[0].Engine == engineOption){ return "selected=\"selected\""; }else{ - if(versionList[0].Engine==122 && engineOption==2){ + if((versionList[0].Engine==122 || versionList[0].Engine==37) && engineOption==2){ return "selected=\"selected\""; } - if(versionList[0].Engine==121 && engineOption==1){ + if((versionList[0].Engine==121 || versionList[0].Engine==38) && engineOption==1){ return "selected=\"selected\""; } } diff --git a/templates/repo/modelmanage/showinfo.tmpl b/templates/repo/modelmanage/showinfo.tmpl index e18c6bd99..4cc0305ed 100644 --- a/templates/repo/modelmanage/showinfo.tmpl +++ b/templates/repo/modelmanage/showinfo.tmpl @@ -244,9 +244,9 @@ function loadInfo(){ function getEngineName(model){ if(model.Engine == 0){ return "PyTorch"; - }else if(model.Engine == 1 || model.Engine == 121){ + }else if(model.Engine == 1 || model.Engine == 121 || model.Engine == 38){ return "TensorFlow"; - }else if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35){ + }else if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35 || model.Engine == 37){ return "MindSpore"; }else if(model.Engine == 3){ return "Other";