|
@@ -78,7 +78,7 @@ func saveModelByParameters(jobId string, versionName string, name string, versio |
|
|
cloudType = models.TypeCloudBrainTwo |
|
|
cloudType = models.TypeCloudBrainTwo |
|
|
spec, err := resource.GetCloudbrainSpec(aiTask.ID) |
|
|
spec, err := resource.GetCloudbrainSpec(aiTask.ID) |
|
|
if err == nil { |
|
|
if err == nil { |
|
|
flaverName := "NPU: " + fmt.Sprint(spec.AccCardsNum) + "*" + spec.AccCardType + ",CPU: " + fmt.Sprint(spec.CpuCores) + "," + ctx.Tr("cloudbrain.memory") + ": " + fmt.Sprint(spec.MemGiB) + "GB," + ctx.Tr("cloudbrain.shared_memory") + ": " + fmt.Sprint(spec.ShareMemGiB) + "GB" |
|
|
|
|
|
|
|
|
flaverName := "NPU: " + fmt.Sprint(spec.AccCardsNum) + "*" + spec.AccCardType + ",CPU: " + fmt.Sprint(spec.CpuCores) + "," + ctx.Tr("cloudbrain.gpu_memory") + ": " + fmt.Sprint(spec.GPUMemGiB) + "GB," + ctx.Tr("cloudbrain.memory") + ": " + fmt.Sprint(spec.MemGiB) + "GB" |
|
|
aiTask.FlavorName = flaverName |
|
|
aiTask.FlavorName = flaverName |
|
|
} |
|
|
} |
|
|
} else if aiTask.ComputeResource == models.GPUResource { |
|
|
} else if aiTask.ComputeResource == models.GPUResource { |
|
|