Commit 234acd42 by vincent

fixed quantization script, pull bytes out of Float32Array when quantization is skipped

parent d872a149
...@@ -58,7 +58,7 @@ ...@@ -58,7 +58,7 @@
} }
if (isSkipQuantization) { if (isSkipQuantization) {
quantizedTensorArrays.push(tensor.dataSync()) quantizedTensorArrays.push(new Uint8Array(tensor.dataSync().buffer))
weightEntries.push(weightEntry) weightEntries.push(weightEntry)
return return
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment