无法预测使用已加载的FrozenModel

Question

我从tfjs-examples/mobilenet/复制了代码，并尝试运行自己的冻结模型，该模型已加载，但是在尝试使用predict方法时会出错。

我正在使用0.14.2版的tfjs和71.0.3578.98版的Google Chrome

我使用了mobilenet回购中显示的tfjs-examples示例，并通过yarn watch启动了服务器。

第二，我成功加载了自己的FrozenModel。

但是当我使用加载模型的predict方法并输入正确形状时，它显示以下错误：

ERROR: 0:163: 'updates' :  left of '[' is not of type array, matrix, or vecto

我只是对mobilenet example中的原始index.js进行了一些修改，脚本如下所示：

import * as tf from '@tensorflow/tfjs';

const MODEL_URL = 'path_to_tensorflowjs_model.pb';
const WEIGHTS_URL = 'path_to_weights_manifest.json';

let gan;
const ganDemo = async () => {
  status('Loading model...');

  gan = await tf.loadFrozenModel(MODEL_URL, WEIGHTS_URL);
  gan.predict(tf.zeros([1, 3, 450, 300])).dispose(); # error here

  ...

我确保模型加载成功，并且输入的形状正确（我有意尝试了其他形状，如果形状不正确，则会引发另一个错误）

任何建议都值得赞赏。

Answer 1

您使用的tfjs npm是什么版本？ 您可以尝试使用最新版本的v1.0.0-alpha2或v0.15.1吗？

有一个bug fix与sparseToDense操作有关。