|
|
@@ -237,9 +237,18 @@ function loadInfo(){ |
|
|
|
return "PyTorch"; |
|
|
|
}else if(model.Engine == 1 || model.Engine == 121){ |
|
|
|
return "TensorFlow"; |
|
|
|
}else if(model.Engine == 2 || model.Engine == 122){ |
|
|
|
}else if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35){ |
|
|
|
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" |
|
|
|
} |
|
|
|
} |
|
|
|