intel / webml-polyfill

Deprecated, the Web Neural Network Polyfill project has been moved to https://github.com/webmachinelearning/webnn-polyfill
Apache License 2.0
161 stars 42 forks source link

[WASM] Check if TF.js can be upgraded to 2.0 in WASM backend #1313

Closed akineeic closed 4 years ago

NALLEIN commented 4 years ago

Upgrade Tensorflow.js to the latest 2.0.1 version, it should be noted that Tensorflow.js has separated tfjs-backend-webgl from tfjs-core

    "@tensorflow/tfjs-backend-wasm": "2.0.1",
    "@tensorflow/tfjs-backend-webgl": "2.0.1",
    "@tensorflow/tfjs-core": "2.0.1",

The RegressionChecker test results are as follows: image No regression occurs, and the model in the example can run normally.

ibelem commented 4 years ago

Good to know, could you please submit a PR for upgrading the backends? @NALLEIN Thanks!