| @@ -237,9 +237,18 @@ function loadInfo(){ | |||||
| return "PyTorch"; | return "PyTorch"; | ||||
| }else if(model.Engine == 1 || model.Engine == 121){ | }else if(model.Engine == 1 || model.Engine == 121){ | ||||
| return "TensorFlow"; | return "TensorFlow"; | ||||
| }else if(model.Engine == 2 || model.Engine == 122){ | |||||
| }else if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35){ | |||||
| return "MindSpore"; | return "MindSpore"; | ||||
| }else{ | |||||
| }else if(model.Engine == 3){ | |||||
| return "Other"; | |||||
| }else if(model.Engine == 4){ | |||||
| return "PaddlePaddle"; | |||||
| }else if(model.Engine == 5){ | |||||
| return "OneFlow"; | |||||
| }else if(model.Engine == 6){ | |||||
| return "MXNet"; | |||||
| } | |||||
| else{ | |||||
| return "Other" | return "Other" | ||||
| } | } | ||||
| } | } | ||||