提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

routers/repo/ai_model_manage.go

@@ -66,6 +66,14 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 			return err
 		}
 	} else if cloudType == models.TypeCloudBrainOne {
+		var ResourceSpecs *models.ResourceSpecs
+		json.Unmarshal([]byte(setting.ResourceSpecs), &ResourceSpecs)
+		for _, tmp := range ResourceSpecs.ResourceSpec {
+			if tmp.Id == aiTask.ResourceSpecId {
+				flaverName := ctx.Tr("cloudbrain.gpu_num") + ":" + fmt.Sprint(tmp.GpuNum) + ctx.Tr("cloudbrain.cpu_num") + ":" + fmt.Sprint(tmp.CpuNum) + ctx.Tr("cloudbrain.memory") + ":" + fmt.Sprint(tmp.MemMiB) + ctx.Tr("cloudbrain.shared_memory") + ":" + fmt.Sprint(tmp.ShareMemMiB)
+				aiTask.FlavorName = flaverName
+			}
+		}
 		modelPath, modelSize, err = downloadModelFromCloudBrainOne(id, aiTask.JobName, "", aiTask.TrainUrl)
 		if err != nil {
 			log.Info("download model from CloudBrainOne faild." + err.Error())

templates/repo/modelmanage/showinfo.tmpl

@@ -191,7 +191,7 @@
                     
                 </div>
             </div>
-            
+            <div id="gpu_resource" style="display:none"> {{$.i18n.Tr "cloudbrain.gpu_num"}}：{GpuNum}，{{$.i18n.Tr "cloudbrain.cpu_num"}}：{CpuNum}，{{$.i18n.Tr "cloudbrain.memory"}}(MB)：{MemMiB}，{{$.i18n.Tr "cloudbrain.shared_memory"}}(MB)：{ShareMemMiB}</div>
         </div>
     </div>
 </div>