Tensorflow.js使用节点保存模型

Posted 2021-04-30

我想使用此功能从node.js保存经过训练的模型

async function tfModelExperiment(model) {
  try {
    let tsModelTraining = await model.save('file:///tmp/my-model-1');
  } 
  catch (error) {
    console.log(error);
  }
}

但是保存模型时会返回

（node：23756）UnhandledPromiseRejectionWarning：错误：找不到任何保存URL'file：/// tmp / my-model-1'的处理程序]

我发现另一个人正在为这个问题而苦苦挣扎，但通过添加来解决此问题>

const tf = require('@tensorflow/tfjs');
我已经拥有了，我已经尝试将目录更改为我的主目录，但这不能解决问题，也不能以sudo的身份运行它，我会怎么做？

软件

我正在将Ubuntu TensorFlow.js软件包（0.13.0）与npm一起安装的Ubuntu Ubuntu 18.04.1 LTS使用]

编辑：

应注意，我尝试过

import * as tf from '@tensorflow/tfjs';
import '@tensorflow/tfjs-node';
如此处提供的（https://github.com/caisq/tfjs-node，返回
TypeError：tf.sequential不是函数在file：///home/sjors/node.mjs：7：18在ModuleJob.run（内部/加载程序/ModuleJob.js:94:14）在

我已经尝试：

const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
与之前返回相同的UnhandledPromiseRejectionWarning错误

我想使用此函数从node.js保存经过训练的模型异步函数tfModelExperiment（model）{试试{let tsModelTraining = await model.save（'file：/// tmp / my-model-1'）; } catch（...

答案

我现在在github上的tfjs的帮助下工作了。

基本上，您只需要安装tfjs-node依赖项：

另一答案

您的异步函数读取其中的await行并执行JS promise。一个诺言是JS编译器执行一段远程代码并确保异步函数将来会向它传递一个值（因此起名为诺言）。

另一答案

我也遇到同样的问题“找不到URL的任何保存处理程序”

另一答案

我使用了npm install -g @tensorflow/tfjs-node 中的npm list -g --depth=0 /usr/local/lib ├── @tensorflow/tfjs-node@0.1.21 ├── express@4.16.4 ├── npm@6.4.1 └── socket.io@2.1.1 ，并成功。