|
|
|
@@ -204,7 +204,7 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri |
|
|
|
}, |
|
|
|
Spec: &modelarts.SpecsC{ |
|
|
|
Resource: &modelarts.ResourceCreateTraining{ |
|
|
|
FlavorId: "modelarts.kat1.xlarge", |
|
|
|
FlavorId: resourceId, |
|
|
|
NodeCount: 1, |
|
|
|
}, |
|
|
|
}, |
|
|
|
@@ -647,7 +647,13 @@ func (m *ModelArtsLink) Execute(ctx context.Context, option *option.AiOption, mo |
|
|
|
if ascendNum == v.FlavorInfo.Npu.UnitNum { |
|
|
|
option.ResourceId = v.FlavorId |
|
|
|
break |
|
|
|
} else if ascendNum == 3 { |
|
|
|
} else if ascendNum <= 1 { |
|
|
|
option.ResourceId = "modelarts.kat1.xlarge" |
|
|
|
break |
|
|
|
} else if ascendNum == 2 { |
|
|
|
option.ResourceId = "modelarts.kat1.2xlarge" |
|
|
|
break |
|
|
|
} else if ascendNum > 2 && ascendNum <= 4 { |
|
|
|
option.ResourceId = "modelarts.kat1.4xlarge" |
|
|
|
break |
|
|
|
} else if ascendNum >= 5 && ascendNum <= 8 { |
|
|
|
|