A TensorFlow integration for deeplearn.js which allows to load tensorflow checkpoint files directly.
For NPM use npm install deeplearn-tensorflow
in your project folder.
For YARN use yarn add deeplearn-tensorflow
in your project folder.
Fork, clone, download the source on GitHub to get the latest version like
git clone https://github.com/google/deeplearn-tensorflow.git
.
import {NDArray, ...} from 'deeplearn';
import {TensorflowLoader} from 'deeplearn-tensorflow';
const tensorflowReader = new TensorflowLoader(NDArray);
tensorflowReader.loadRemoteFiles('data/model.ckpt-999').then((vars) => {
...
});
For an demo implementation, please check DEMO.md.
This is not an official Google product.
Apache License, Version 2.0 - http://www.apache.org/licenses/LICENSE-2.0