Browse Source

Merge pull request '屏蔽智算网络的模型创建及支持MindSpore1.5的模型框架显示。' (#2330) from zouap into V20220616.patch

Reviewed-on: https://git.openi.org.cn/OpenI/aiforge/pulls/2330
Reviewed-by: lewis <747342561@qq.com>
V20220616.patch
lewis 3 years ago
parent
commit
bc9c7a2992
3 changed files with 5 additions and 2 deletions
  1. +3
    -0
      models/cloudbrain.go
  2. +1
    -1
      templates/repo/modelarts/trainjob/show.tmpl
  3. +1
    -1
      web_src/js/components/Model.vue

+ 3
- 0
models/cloudbrain.go View File

@@ -1329,6 +1329,9 @@ func QueryModelTrainJobList(repoId int64) ([]*CloudbrainInfo, int, error) {
cond = cond.And( cond = cond.And(
builder.Eq{"job_type": "TRAIN"}, builder.Eq{"job_type": "TRAIN"},
) )
cond = cond.And(
builder.In("type", 0, 1),
)


cloudbrains := make([]*CloudbrainInfo, 0) cloudbrains := make([]*CloudbrainInfo, 0)
if err := sess.Select("job_id,display_job_name").Table(&Cloudbrain{}).Where(cond).OrderBy("created_unix DESC"). if err := sess.Select("job_id,display_job_name").Table(&Cloudbrain{}).Where(cond).OrderBy("created_unix DESC").


+ 1
- 1
templates/repo/modelarts/trainjob/show.tmpl View File

@@ -675,7 +675,7 @@
$('#JobName').val(obj.DisplayJobName).addClass('model_disabled') $('#JobName').val(obj.DisplayJobName).addClass('model_disabled')
$('input[name="JobId"]').val(obj.JobID) $('input[name="JobId"]').val(obj.JobID)
$('input[name="VersionName"]').val(obj.VersionName).addClass('model_disabled') $('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_name"]').val("MindSpore").addClass('model_disabled');
$('input[name="Engine"]').val(2); $('input[name="Engine"]').val(2);
} }


+ 1
- 1
web_src/js/components/Model.vue View File

@@ -353,7 +353,7 @@ export default {
return "Pytorch"; return "Pytorch";
}else if(model.Engine == 1 || model.Engine == 121){ }else if(model.Engine == 1 || model.Engine == 121){
return "TensorFlow"; return "TensorFlow";
}else if(model.Engine == 2 || model.Engine == 122){
}else if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35){
return "MindSpore"; return "MindSpore";
}else{ }else{
return "Other" return "Other"


Loading…
Cancel
Save