+
82
-

nodejs可以运行tensorflow吗?

nodejs可以运行tensorflow吗?

网友回复

+
1
-

可以

npm install @tensorflow/tfjs @tensorflow/tfjs-node
或者
npm install @tensorflow/tfjs @tensorflow/tfjs-node-gpu

就可以直接安装nodejs版本的tensorflow,gpu表示使用gpu进行训练的版本

+
0
-

const saveResult = await model.save('localstorage://my-model-1');

保存到浏览器localstorage中,key名为my-model-1

那么怎么加载?

const model = await tf.loadLayersModel('localstorage://my-model-1');

还可以在直接下载model

await model.save('downloads://my-model');

这会让浏览器下载模型文件至用户的机器上,并生成两个文件: 1. 一个名为 [my-model].json 的 JSON 文件,它包含了模型的拓扑结构和下面将要介绍的权重文件的引用。 2. 一个二进制文件,其中包含名为 [my-model].weights.bin 的权重值。 您可以更换 [my-model] 的名称以获得一个不同的名称的文件。 由于.json使用相对路径指向 .bin,所以两个文件需要被安放在同一个文件夹中。
我知道答案,我要回答