How to use the @tensorflow/tfjs-node-gpu.loadLayersModel function in @tensorflow/tfjs-node-gpu
To help you get started, we’ve selected a few @tensorflow/tfjs-node-gpu examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
bobiblazeski / js-gym / dist / agents.gpu.js View on Github 
async load(infix) {
const aPath = `${SAVE_METHOD}${infix}_actor/model.json`;
const cPath = `${SAVE_METHOD}${infix}_critic/model.json`;
this.actor = await tf.loadLayersModel(aPath);
this.critic = await tf.loadLayersModel(cPath);
hardUpdate(this.actor, this.actorTarget);
hardUpdate(this.critic, this.criticTarget);
}
}async load(infix) {
const aPath = `${SAVE_METHOD}${infix}_actor/model.json`;
const cPath = `${SAVE_METHOD}${infix}_critic/model.json`;
this.actor = await tf.loadLayersModel(aPath);
this.critic = await tf.loadLayersModel(cPath);
hardUpdate(this.actor, this.actorTarget);
hardUpdate(this.critic, this.criticTarget);
}
}@tensorflow/tfjs-node-gpu
For information please refer [here](https://github.com/tensorflow/tfjs/blob/master/tfjs-node/README.md).
Package Health Score
81 / 100Popular @tensorflow/tfjs-node-gpu functions
- @tensorflow/tfjs-node-gpu.add
- @tensorflow/tfjs-node-gpu.cast
- @tensorflow/tfjs-node-gpu.concat
- @tensorflow/tfjs-node-gpu.data
- @tensorflow/tfjs-node-gpu.dispose
- @tensorflow/tfjs-node-gpu.expandDims
- @tensorflow/tfjs-node-gpu.input
- @tensorflow/tfjs-node-gpu.layers
- @tensorflow/tfjs-node-gpu.loadLayersModel
- @tensorflow/tfjs-node-gpu.memory
- @tensorflow/tfjs-node-gpu.model
- @tensorflow/tfjs-node-gpu.mul
- @tensorflow/tfjs-node-gpu.node
- @tensorflow/tfjs-node-gpu.randomUniform
- @tensorflow/tfjs-node-gpu.sequential
- @tensorflow/tfjs-node-gpu.tensor
- @tensorflow/tfjs-node-gpu.tidy
- @tensorflow/tfjs-node-gpu.train
- @tensorflow/tfjs-node-gpu.util
- @tensorflow/tfjs-node-gpu.zeros