diff --git a/models/cloudbrain.go b/models/cloudbrain.go index 4840b78c0..8ec088b28 100755 --- a/models/cloudbrain.go +++ b/models/cloudbrain.go @@ -1329,6 +1329,9 @@ func QueryModelTrainJobList(repoId int64) ([]*CloudbrainInfo, int, error) { cond = cond.And( builder.Eq{"job_type": "TRAIN"}, ) + cond = cond.And( + builder.In("type", 0, 1), + ) cloudbrains := make([]*CloudbrainInfo, 0) if err := sess.Select("job_id,display_job_name").Table(&Cloudbrain{}).Where(cond).OrderBy("created_unix DESC"). diff --git a/templates/repo/modelarts/trainjob/show.tmpl b/templates/repo/modelarts/trainjob/show.tmpl index 674e7e859..615fc3030 100755 --- a/templates/repo/modelarts/trainjob/show.tmpl +++ b/templates/repo/modelarts/trainjob/show.tmpl @@ -675,7 +675,7 @@ $('#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){ + if(obj.EngineID ==122 || obj.EngineID ==35){ $('input[name="Engine_name"]').val("MindSpore").addClass('model_disabled'); $('input[name="Engine"]').val(2); } diff --git a/web_src/js/components/Model.vue b/web_src/js/components/Model.vue index 801ca8805..ed409c0d2 100644 --- a/web_src/js/components/Model.vue +++ b/web_src/js/components/Model.vue @@ -353,7 +353,7 @@ export default { 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{ return "Other"