|
|
@@ -322,7 +322,7 @@ |
|
|
|
loadCheckpointList(dataID).then((res)=>{ |
|
|
|
res.forEach(element => { |
|
|
|
const ckptSuffix = element.FileName.split(".") |
|
|
|
const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7'] |
|
|
|
const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7','pdparams','onnx','pbtxt','keras','mlmodel','cfg','pt'] |
|
|
|
if(!element.IsDir && loadCheckpointFile.includes(ckptSuffix[ckptSuffix.length-1])){ |
|
|
|
html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>` |
|
|
|
} |
|
|
|