diff options
| -rw-r--r-- | cnn_v3/tools/tester.js | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/cnn_v3/tools/tester.js b/cnn_v3/tools/tester.js index 81c869d..69f358b 100644 --- a/cnn_v3/tools/tester.js +++ b/cnn_v3/tools/tester.js @@ -106,7 +106,7 @@ class CNNv3Tester { if (u32.length < TOTAL_U32) throw new Error(`Too small: ${u32.length} u32, need ${TOTAL_U32}`); const layers = [ {n:'enc0',off:ENC0_OFF,cnt:724},{n:'enc1',off:ENC1_OFF,cnt:296}, - {n:'bn', off:BN_OFF, cnt: 72},{n:'dec1',off:DEC1_OFF,cnt:580}, + {n:'bn', off:BN_OFF, cnt:584},{n:'dec1',off:DEC1_OFF,cnt:580}, {n:'dec0',off:DEC0_OFF,cnt:292}, ]; let html=`<div style="margin-bottom:7px"><b>Size:</b> ${(buf.byteLength/1024).toFixed(1)} KB <b>Weights:</b> ${TOTAL_F16} f16</div> @@ -387,7 +387,7 @@ class CNNv3Tester { // Store for layer viz & redisplay this.destroyLayerTex(); - this.layerTextures={feat0:f0,feat1:f1,enc0:e0,enc1:e1,bn,dec1:d1,output:ot}; + this.layerTextures={feat0:f0,feat1:f1,enc0:e0,enc1:e1,bn,dec1:d1,dec0:ot}; this.lastResult={ot,itex:this.inputTex,uDp,dispPL:this.getDisp(),w,h}; this.updateVizPanel(); this.refreshZoom(); @@ -446,7 +446,7 @@ class CNNv3Tester { {id:'enc1', lbl:'Enc1', t:'u32',nch:8, ch:['c0','c1','c2','c3','c4','c5','c6','c7']}, {id:'bn', lbl:'BN', t:'u32',nch:8, ch:['c0','c1','c2','c3','c4','c5','c6','c7']}, {id:'dec1', lbl:'Dec1', t:'f32',nch:4, ch:['c0','c1','c2','c3']}, - {id:'output',lbl:'Output', t:'f32',nch:4, ch:['R','G','B','A']}, + {id:'dec0', lbl:'Dec0', t:'f32',nch:4, ch:['R','G','B','A']}, ]; this.vizDefs=DEFS; const panel=document.getElementById('layerViz'); @@ -454,7 +454,7 @@ class CNNv3Tester { for(const d of DEFS) html+=`<button id="vb_${d.id}" onclick="tester.vizLayer('${d.id}')">${d.lbl}</button>`; html+='</div><div class="chgrid" id="chgrid"></div>'; panel.innerHTML=html; - this.vizLayer('output'); + this.vizLayer('dec0'); } async vizLayer(id) { |