Browse Source

提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>
pull/2480/head
zouap 3 years ago
parent
commit
0d7fa4011d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      routers/repo/ai_model_convert.go

+ 1
- 1
routers/repo/ai_model_convert.go View File

@@ -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


Loading…
Cancel
Save