1 changed files with 12 additions and 1 deletions
Unified View
Diff Options
+ 12
- 1
routers/repo/ai_model_convert.go
View File
| @@ -52,6 +52,9 @@ const ( | |||||
| CONVERT_FORMAT_ONNX = 0 | CONVERT_FORMAT_ONNX = 0 | ||||
| CONVERT_FORMAT_TRT = 1 | CONVERT_FORMAT_TRT = 1 | ||||
| NetOutputFormat_FP32 = 0 | |||||
| NetOutputFormat_FP16 = 1 | |||||
| NPU_MINDSPORE_IMAGE_ID = 35 | NPU_MINDSPORE_IMAGE_ID = 35 | ||||
| NPU_TENSORFLOW_IMAGE_ID = 121 | NPU_TENSORFLOW_IMAGE_ID = 121 | ||||
| @@ -460,7 +463,15 @@ func getGpuModelConvertCommand(name string, modelFile string, modelConvert *mode | |||||
| h = intputshape[2] | h = intputshape[2] | ||||
| w = intputshape[3] | w = intputshape[3] | ||||
| } | } | ||||
| command += "python3 /code/" + bootfile + " --model " + modelFile + " --n " + n + " --c " + c + " --h " + h + " --w " + w + " > " + ModelMountPath + "/" + name + "-" + LogFile | |||||
| command += "python3 /code/" + bootfile + " --model " + modelFile + " --n " + n + " --c " + c + " --h " + h + " --w " + w | |||||
| if modelConvert.DestFormat == CONVERT_FORMAT_TRT { | |||||
| if modelConvert.NetOutputFormat == NetOutputFormat_FP16 { | |||||
| command += " --fp16 True" | |||||
| } else { | |||||
| command += " --fp16 False" | |||||
| } | |||||
| } | |||||
| command += " > " + ModelMountPath + "/" + name + "-" + LogFile | |||||
| return command | return command | ||||
| } | } | ||||