From a027ec6867ae726ea6f90815d735f11a131377cb Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Wed, 15 Jun 2022 09:53:05 +0800
Subject: [PATCH 1/3] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 options/locale/locale_en-US.ini          | 1 +
 options/locale/locale_zh-CN.ini          | 1 +
 templates/repo/modelmanage/showinfo.tmpl | 2 +-
 web_src/js/components/Model.vue          | 2 +-
 4 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/options/locale/locale_en-US.ini b/options/locale/locale_en-US.ini
index 016aa4951..dc004c45e 100755
--- a/options/locale/locale_en-US.ini
+++ b/options/locale/locale_en-US.ini
@@ -1092,6 +1092,7 @@ modelarts.train_job.fast_parameter_setting_config_link=fast_parameter_setting_co
 modelarts.train_job.frames=frames
 modelarts.train_job.algorithm_origin=Algorithm Origin
 modelarts.train_job.AI_driver=AI Engine
+modelarts.train_job.AI_Engine=AI Engine
 modelarts.train_job.start_file=Start File
 modelarts.train_job.boot_file_helper=The startup file is the entry file that your program executes, and it must be a file ending in .py
 modelarts.train_job.dataset=Dataset
diff --git a/options/locale/locale_zh-CN.ini b/options/locale/locale_zh-CN.ini
index 440f6746a..36fdc804a 100755
--- a/options/locale/locale_zh-CN.ini
+++ b/options/locale/locale_zh-CN.ini
@@ -1102,6 +1102,7 @@ modelarts.train_job.fast_parameter_setting_config_link=这里
 modelarts.train_job.frames=常用框架
 modelarts.train_job.algorithm_origin=算法来源
 modelarts.train_job.AI_driver=AI引擎
+modelarts.train_job.AI_Engine=模型框架
 modelarts.train_job.start_file=启动文件
 modelarts.train_job.boot_file_helper=启动文件是您程序执行的入口文件，必须是以.py结尾的文件。比如train.py、main.py、example/train.py、case/main.py。
 modelarts.train_job.boot_file_place=填写启动文件路径，默认为train.py
diff --git a/templates/repo/modelmanage/showinfo.tmpl b/templates/repo/modelmanage/showinfo.tmpl
index 0de4e043b..bb4807c07 100644
--- a/templates/repo/modelmanage/showinfo.tmpl
+++ b/templates/repo/modelmanage/showinfo.tmpl
@@ -141,7 +141,7 @@
                                 <td class="ti-text-form-content word-elipsis"><span id="Parameters" title=""></span></td>
                             </tr>
                             <tr>
-                                <td class="ti-text-form-label text-width80">{{$.i18n.Tr "repo.modelarts.train_job.AI_driver"}}</td>
+                                <td class="ti-text-form-label text-width80">{{$.i18n.Tr "repo.modelarts.train_job.AI_Engine"}}</td>
                                 <td class="ti-text-form-content word-elipsis"><span id="EngineName" title=""></span></td>
                             </tr>
                             <tr>
diff --git a/web_src/js/components/Model.vue b/web_src/js/components/Model.vue
index d9672214c..801ca8805 100644
--- a/web_src/js/components/Model.vue
+++ b/web_src/js/components/Model.vue
@@ -60,7 +60,7 @@
                         </el-table-column>
                         <el-table-column
                             prop="EngineName"
-                            label="AI引擎"
+                            label="模型框架"
                             align="center"
                             min-width="8.5%"
                             >

From e1acb6777b1478a08ce258332ade5fdb78802207 Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Wed, 15 Jun 2022 10:16:52 +0800
Subject: [PATCH 2/3] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 templates/repo/modelmanage/index.tmpl | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/templates/repo/modelmanage/index.tmpl b/templates/repo/modelmanage/index.tmpl
index 0c5cd5547..ec96606b9 100644
--- a/templates/repo/modelmanage/index.tmpl
+++ b/templates/repo/modelmanage/index.tmpl
@@ -1,5 +1,10 @@
 <!-- 头部导航栏 -->
 {{template "base/head" .}}
+<style>
+.defaulttext{
+    color: rgba(0,0,0,.87)
+}
+</style>
 <!-- 弹窗 -->
 <div id="mask">
     <div id="loadingPage">
@@ -135,7 +140,7 @@
                     <label>模型框架</label>
                     <div class="ui dropdown selection search width70" id="choice_Engine">
                             <input type="hidden" id="Engine" name="Engine" required>
-                            <div class="default text">选择模型框架</div>
+                            <div class="default defaulttext">选择模型框架</div>
                             <i class="dropdown icon"></i>
                             <div class="menu" id="job-Engine">
                                 
@@ -316,7 +321,7 @@
             }
             itemHtml += "<option class=\"item\" data-value=\"3\">Other</option>"
             
-            $('#choice_Engine .default.text').text(selectedText)
+            $('#choice_Engine .default.defaulttext').text(selectedText)
             $('#choice_Engine input[name="Engine"]').val(selectedValue)
             $("#job-Engine").append(itemHtml);
             $("#choice_Engine").addClass('disabled')
@@ -325,7 +330,7 @@
             itemHtml += "<option class=\"item\" data-value=\"1\">TensorFlow</option>"
             itemHtml += "<option class=\"item\" data-value=\"2\">MindSpore</option>"
             itemHtml += "<option class=\"item\" data-value=\"3\">Other</option>"
-            $('#choice_Engine .default.text').text("Pytorch");
+            $('#choice_Engine .default.defaulttext').text("Pytorch");
             $('#choice_Engine input[name="Engine"]').val(0)
             $("#job-Engine").append(itemHtml);
             $("#choice_Engine").removeClass('disabled');

From 5264394e34842cfd769cabed6d1ddebb7882d43e Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Wed, 15 Jun 2022 10:20:04 +0800
Subject: [PATCH 3/3] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 templates/repo/modelmanage/showinfo.tmpl | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

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
@@ -233,12 +233,24 @@ function loadInfo(){
         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)