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