|
|
@@ -532,7 +532,7 @@ |
|
|
|
} |
|
|
|
} |
|
|
|
function isModel(filename){ |
|
|
|
var postfix=[".pth",".pkl",".onnx",".mindir",".ckpt",".pb",".pdmodel","pdparams",".params",".json"]; |
|
|
|
var postfix=[".pth",".pkl",".onnx",".mindir",".ckpt",".pb",".pdmodel","pdiparams",".params",".json"]; |
|
|
|
for(var i =0; i<postfix.length;i++){ |
|
|
|
if(filename.substring(filename.length-postfix[i].length)==postfix[i]){ |
|
|
|
return true; |
|
|
|