TensorFlow.js tf.GraphModel 类 .dispose() 方法介绍

在TensorFlow.js中，tf.GraphModel是一种用于加载预训练模型并进行推理的类。当你完成了模型的所有操作时，你应该释放它，以便回收内存。这时就需要使用到 dispose() 方法。

dispose() 方法

tf.GraphModel 类的 dispose() 方法用于释放已加载的模型。使用 dispose() 方法后，该模型将不再可用，释放其引用可让浏览器垃圾回收器更轻松地清除它所占用的内存。方法签名如下：

dispose(): void

使用方法

要释放 tf.GraphModel 对象，只需调用 dispose() 方法，如下所示：

// 加载模型
const model = await tf.loadGraphModel('model.json');

// 使用模型

// 释放模型
model.dispose();

调用 dispose() 方法后，该模型将被销毁，不再可用。如果你尝试使用该模型进行操作，则会报错。

总结

使用 TensorFlow.js tf.GraphModel 类的 dispose() 方法可以轻松释放已加载的模型，回收内存。在不需要使用预训练模型时，及时释放模型可以提高浏览器资源的利用效率，同时也有助于减少浏览器的崩溃情况。