| ``` | |
| const tf = require('@tensorflow/tfjs'); | |
| const tfnode = require('@tensorflow/tfjs-node'); | |
| async function loadModel(){ | |
| const handler = tfnode.io.fileSystem('checkpoint/model.json'); | |
| const model = await tf.loadGraphModel(handler); | |
| console.log("Model loaded") | |
| const zeros = tf.randomNormal([8, 3, 224, 224]); | |
| const out = model.predict(zeros); | |
| console.log(out); | |
| out.print(); | |
| } | |
| loadModel(); | |
| ``` | |