问题标签 [tfjs-node]

我在 Keras 中创建了一个模型并将其转换为 Tensorflow.js 模型并将其加载到我的 node.js 项目中。现在我想从 Tensorflow.js 中的这个模型中得到预测。我已经弄清楚如何加载单个图像：

这将创建一个 shape 的张量(imageWidth, imageHeight, 3)。但我想将一批图像加载到一个 shape 的张量中(batchNumber, imageWidth, imageHeight, 3)。

我该怎么做呢？

我的 tfjs-node-gpu 代码在 GKE 上的 NVIDIA p4 上运行良好（并在浏览器中使用 WebGL），但在 v100 和 t4 上却失败了。

节点在我的热身中的第一个预测调用中崩溃。我正在使用 128x128 的小图块来预测使用 Idealo-gans 的 4 倍图像放大。v100 初始化很好，显示为 nvidia_smi，显示为 TF 设备，NUMA 的东西都很好。它只是让我的节点快递服务器崩溃。我无法找到崩溃堆栈，因为这是在 Docker 容器中启动的，并且我上次尝试从 stderr 记录崩溃失败。

我已经尝试过最新的 tfjs-node-gpu 3.0 和 2.8.5。GKE 配置为安装 NV 驱动程序，当前为 410.104 和 CUDA 10.0。

我尝试启用调试模式，并在我的预热函数中传递{verbose: true}给失败的调用。model.predict()两者都没有向预热调用添加任何输出，这很奇怪，因为我确实在实际的非预热调用中看到了输出model.predict()

关于如何进一步调试的任何建议？

尝试在 max m1 芯片中使用 tfjs 运行基本入门示例并获得非法参数：4 错误。Mac m1 芯片有什么特别需要做的吗？任何帮助

我不知道这里发生了什么。我已经看到了电子的解决方案，但它不适用于这种情况。

我的节点 -v 是 v10.16.3

我的 package.json 是：

我的代码昨天正在运行。现在，当我运行时node bot.js，它给了我以下信息：

当我跑步时，他们要求我跑步：npm rebuild @tensorflow/tfjs-node build-addon-from-source，我得到：

提前致谢。

我一直在尝试使用 tfjs 建立一个简单的强化学习示例。但是，在尝试训练模型时，我遇到了以下错误：

我建立了如下模型：

训练是由一个计算一些例子的 Q 值的函数完成的：

这似乎是为 fit 函数提供示例的正确方法，因为在记录模型时，最后一个密集层的形状是正确的：

dense_Dense5 形状的日志

然而，它会导致上面显示的错误，而不是预期的形状 [3,4] 它检查形状 [,1]。我真的不明白这个形状是从哪里突然来的，非常感谢一些帮助！

为了获得更好的概览，您可以从其 Github 存储库中简单地查看/签出整个项目：

Github 回购

有问题的 tensorflow 代码位于 AI 文件夹中。

编辑：

y提供模型摘要以及提供的张量形状的一些信息model.fit(x,y)：

我正在通过 Node.js (@tensorflow/tfjs-node) 使用 TensorFlow.js 的 TensorFlow 后端在 Node.js 中运行预训练的 tf 模型，但是当我尝试将数据传递给模型时出现错误。
模型的输入如下获得tf.node.getMetaGraphsFromSavedModel(path)：

加载模型工作正常，但是当我尝试运行以下代码时：

我收到以下错误消息：

错误：会话无法运行并出现错误：audio_id:0 已输入和提取。

看起来audio_id既是输入又是输出，并且指的是同一个占位符。我该如何解决这个错误？

在我的本地，它运行良好 - 为了推动 Heroku，我确保：

1 - 服务器的名称是index.js

2 - 我的config.js变量被推送到构建中（这是一个私人仓库，所以这不是问题）。

3 - 我package.json的列表"start": "node index.js"

当我在命令行上部署时，构建成功。当我打开 Heroku 的日志时，会发生以下情况：

似乎已部署软件包？还是 Tensorflow.js 有问题？我真的不确定。谢谢你的协助。

我正在尝试根据此文档使用我自己的训练模型执行离线识别：https ://github.com/tensorflow/tfjs-models/tree/master/speech-commands

我遇到了与https://github.com/tensorflow/tfjs/issues/3820描述的相同的问题，并且我尝试了从那里建议的所有解决方案，包括 colab（预处理模型）支持https://colab.research.google .com/github/tensorflow/tfjs-models/blob/master/speech-commands/training/browser-fft/training_custom_audio_model_in_python.ipynb#scrollTo=1AjdTru5NnQQ与给定的wav文件配合良好，但在使用我的自带wav文件：</p>

有没有办法解决这个问题？

例如，我应该根据给定的 wav 文件修改我的 wav 文件数据吗？但是怎么做？在处理自己的 wav 文件时，我是否错过了预处理过程中的一些重要步骤？或者有没有更简单的方法可以在 javascript 中而不是在 python 中实现这一点？

这是一个后续问题：无法在公司代理后面安装 tfjs-node

是否可以设置一个环境变量来跳过“libtensorflow”的下载并指向 dll 文件？

我正在尝试在 tensorflow.js 中微调 CNN。为此，我想在预训练模型的最后一层添加一个头部。python tensorflow 中的等效代码如下，我们在预训练的效率网络中添加了一个平均池化层。

但是，JavaScript 中的相同代码会导致错误。

打印imgBase.outputs给我们以下结果。imgBase.outputs[0]返回与上述相同的错误。

我们如何获得基础模型的输出，以便将其输入到单独的层中？谢谢。