训练 GAN 模型时 Tensorflow js 后端错误

后端错误"/>

训练 GAN 模型时 Tensorflow js 后端错误

我猜错误出在我整洁的函数中，但不知道在哪里。我在最新版本的 Node.js 和@tensorflow/tfjs-node

    // Train the discriminator
    tf.tidy(() => {
        imageDataset.forEachAsync(async (batch) => {
            const noise = tf.randomNormal([BATCH_SIZE, NOISE_DIM]);
            const generatedImages = generator.predict(noise);
            const realLabels = tf.ones([BATCH_SIZE, 1]);
            const fakeLabels = tf.zeros([BATCH_SIZE, 1]);

            const realLoss = discriminator.trainOnBatch(batch, realLabels);
            const fakeLoss = discriminator.trainOnBatch(generatedImages, fakeLabels);
            const discriminatorLoss = realLoss.add(fakeLoss).div(tf.scalar(2));

            console.log(`Discriminator loss: ${discriminatorLoss}`);
            
            // Clean up the tensors
            tf.dispose([noise, generatedImages, realLabels, fakeLabels]);
        });
    });

    // Train the generator
    tf.tidy(() => {
        const noise = tf.randomNormal([BATCH_SIZE, NOISE_DIM]);
        const fakeLabels = tf.ones([BATCH_SIZE, 1]);

        const generatorLoss = gan.trainOnBatch(noise, fakeLabels);

        console.log(`Generator loss: ${generatorLoss}`);

        // Clean up the tensors
        tf.dispose([noise, fakeLabels]);
    });

    // Save the generator model
    if ((epoch + 1) % 10 === 0) {
        console.log("Saving generator model...");
        generator.save(`file://generator_${epoch + 1}`);
    }
}

    C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:4466
        var srcBackend = info.backend;
                              ^

TypeError: Cannot read properties of undefined (reading 'backend')
    at Engine.moveData (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:4466:31)
    at DataStorage.get (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:232:28)
    at NodeJSKernelBackend.getInputTensorIds (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\dist\nodejs_kernel_backend.js:172:43)
    at NodeJSKernelBackend.executeSingleOutput (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\dist\nodejs_kernel_backend.js:219:73)
    at Object.kernelFunc (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\dist\kernels\Reshape.js:34:27)
    at kernelFunc (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:4651:32)
    at C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:4711:27
    at Engine.scopedRun (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:4516:23)
    at Engine.runKernelFunc (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:4707:14)
    at Engine.runKernel (C:\Users\guill\OneDrive\Bureau\Startup\node_modules\@tensorflow\tfjs-node\node_modules\@tensorflow\tfjs-core\dist\tf-core.node.js:4580:21)

我认为这可能是控制台日志功能，但不是那些，这个错误发生在第 2 个时期而不是之前。我尝试将后端设置为不同的东西，但也无济于事。