|
|
@@ -99,7 +99,7 @@ func SaveModelConvert(ctx *context.Context) { |
|
|
|
UserId: ctx.User.ID, |
|
|
|
} |
|
|
|
models.SaveModelConvert(modelConvert) |
|
|
|
if modelConvert.SrcEngine == PYTORCH_ENGINE || modelConvert.SrcEngine == TENSORFLOW_ENGINE { |
|
|
|
if modelConvert.SrcEngine == PYTORCH_ENGINE { |
|
|
|
err = createGpuTrainJob(modelConvert, ctx, task.Path) |
|
|
|
} else { |
|
|
|
//create npu job |
|
|
|