| @@ -311,15 +311,16 @@ | |||||
| loadCheckpointList(dataID).then((res)=>{ | loadCheckpointList(dataID).then((res)=>{ | ||||
| res.forEach(element => { | res.forEach(element => { | ||||
| const ckptSuffix = element.FileName.split(".") | const ckptSuffix = element.FileName.split(".") | ||||
| if(!element.IsDir && ckptSuffix[ckptSuffix.length-1]==='ckpt'){ | |||||
| const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7'] | |||||
| 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>` | ||||
| } | } | ||||
| }) | }) | ||||
| $('#model_checkpoint').append(html) | $('#model_checkpoint').append(html) | ||||
| $("#select_model_checkpoint").removeClass("loading") | $("#select_model_checkpoint").removeClass("loading") | ||||
| const initVersionText = $('#model_checkpoint div.item:last-child').text() | |||||
| const initVersionValue = $('#model_checkpoint div.item:last-child').data('value') | |||||
| const initVersionText = $('#model_checkpoint div.item:first-child').text() | |||||
| const initVersionValue = $('#model_checkpoint div.item:first-child').data('value') | |||||
| $("#select_model_checkpoint").dropdown('set text',initVersionText) | $("#select_model_checkpoint").dropdown('set text',initVersionText) | ||||
| $("#select_model_checkpoint").dropdown('set value',initVersionValue,initVersionText,$('#model_name_version div.item:first-child')) | $("#select_model_checkpoint").dropdown('set value',initVersionValue,initVersionText,$('#model_name_version div.item:first-child')) | ||||
| }) | }) | ||||