@@ -322,7 +322,7 @@ | |||||
loadCheckpointList(dataID).then((res)=>{ | loadCheckpointList(dataID).then((res)=>{ | ||||
res.forEach(element => { | res.forEach(element => { | ||||
const ckptSuffix = element.FileName.split(".") | 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])){ | if(!element.IsDir && loadCheckpointFile.includes(ckptSuffix[ckptSuffix.length-1])){ | ||||
html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>` | html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>` | ||||
} | } | ||||
@@ -339,7 +339,7 @@ | |||||
loadCheckpointList(dataID).then((res)=>{ | loadCheckpointList(dataID).then((res)=>{ | ||||
res.forEach(element => { | res.forEach(element => { | ||||
const ckptSuffix = element.FileName.split(".") | 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])){ | if(!element.IsDir && loadCheckpointFile.includes(ckptSuffix[ckptSuffix.length-1])){ | ||||
html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>` | html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>` | ||||
} | } | ||||