0

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

inputs: {
      audio_id: [Object],
      mix_spectrogram: [Object],
      mix_stft: [Object],
      waveform: [Object]
    },
    outputs: { accompaniment: [Object], audio_id: [Object], vocals: [Object] }

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

        const inputs = {
            audio_id: tf.tensor(['id']),
            mix_spectrogram: tf.randomNormal([2, 512, 1024, 2]),
            mix_stft: tf.randomNormal([2, 2049, 2]),
            waveform: tf.randomNormal([2, 2])
        };
        const output = model.predict(inputs)

我收到以下错误消息：

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

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

4

0 回答 0