Question

我有一个训练有素的模型，并用const saveResult = await model.save('localstorage://my-model-1');保存了它。

现在，我想重新加载它并再次使用它。所以我想做这样的事情：

async function loadModel() {
    let myModel = tf.sequential();
    myModel = await model.save('downloads://my-model-1')
    console.log(myModel);

    let outputs = myModel.predict([
        tf.tensor2d([[0, 0, 1]])
    ]);
    outputs.print();
}

但是似乎model.save返回了modelArtifactsInfo。那么如何从该对象创建模型？

我关注了this tutorial，但他们并没有真正解释。

那有可能吗？

Answer 1

在本教程中，它说明了一切：

您只需要用保存句柄调用tf.loadModel，它就会返回一个承诺，该承诺可以解析为已加载的模型或引发错误。在您的情况下，句柄为'localstorage://my-model-1'。

const model = await tf.loadModel('localstorage://my-model-1');

Tensorflow.js从下载的文件创建模型

1 个答案: