From 1dc9733e1ed43ffda9ff6ffd3f2520761c6f92dd Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Mon, 4 Jul 2022 14:41:17 +0800
Subject: [PATCH] =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E7=AE=A1=E7=90=86=E5=86=B2?=
 =?UTF-8?q?=E7=AA=81=E8=A7=A3=E5=86=B3=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/grampus/trainjob/show.tmpl | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/templates/repo/grampus/trainjob/show.tmpl b/templates/repo/grampus/trainjob/show.tmpl
index 779097e85..a607221f9 100755
--- a/templates/repo/grampus/trainjob/show.tmpl
+++ b/templates/repo/grampus/trainjob/show.tmpl
@@ -570,7 +570,7 @@
                             <div class="default text">选择模型框架</div>
                             <i class="dropdown icon"></i>
                             <div class="menu" id="job-Engine">
-                                <option class="active item" data-value="0">Pytorch</option>
+                                <option class="active item" data-value="0">PyTorch</option>
                                 <option class="item" data-value="1">TensorFlow</option>
                                 <option class="item" data-value="4">PaddlePaddle</option>
                                 <option class="item" data-value="5">OneFlow</option>
@@ -759,12 +759,17 @@
                     $('input[name="JobId"]').val(obj.JobID)
                     $('input[name="VersionName"]').val("V0001")
                    
-                    $('#choice_Engine .default.text').text("Pytorch");
-                    $('#choice_Engine input[name="Engine"]').val(0)
-                   
-                    $('.ui.dimmer').css({ "background-color": "rgb(136, 136, 136,0.7)" })
-                    createModelName();
-                    loadSelectedModelFile(obj);
+                   if(obj.ComputeResource=="NPU"){
+                        $('#choice_Engine .default.text').text("TensorFlow");
+                        $('#choice_Engine input[name="Engine"]').val(1)
+                   }else{
+                        $('#choice_Engine .default.text').text("PyTorch");
+                        $('#choice_Engine input[name="Engine"]').val(0)
+                   }
+                    
+                   $('.ui.dimmer').css({ "background-color": "rgb(136, 136, 136,0.7)" })
+                   createModelName();
+                   loadSelectedModelFile(obj);
                 },
                 onHide: function () {
                     var cityObj = $("#modelSelectedFile");