Tensorflow.js tf.GraphModel.dispose()用法及代码示例

Tensorflow.js是由Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

.dispose() 函数用于释放权重张量和资源管理器正在使用的内存。

用法:

dispose()

Parameters:

此方法不保存任何参数。

返回值：它返回void。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input elements
const model_Url =
'https://storage.googleapis.com/tfjs-models/savedmodel/mobilenet_v2_1.0_224/model.json';
  
// Calling the loadGraphModel() method
const mymodel = await tf.loadGraphModel(model_Url);
  
// Calling dispose() method
mymodel.dispose();
  
// Printing output
console.log('Model Disposed.');

输出：

Model Disposed.

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input elements
const model_Url =
'https://storage.googleapis.com/tfjs-models/savedmodel/mobilenet_v2_1.0_224/model.json';
  
// Calling the loadGraphModel() method
const mymodel = await tf.loadGraphModel(model_Url);
  
// Defining inputs
const inputs = tf.zeros([1, 224, 224, 3]);
  
// Calling dispose() method
mymodel.dispose();
  
// Calling execute() method and
// Printing output
mymodel.execute(inputs).print();

输出：

An error occured
Cannot read property 'backend' of undefined

在这里，发生了错误并且没有打印输出，因为所述模型已经被处理。因此，execute() 方法无法返回任何输出。

参考： https://js.tensorflow.org/api/latest/#tf.GraphModel.dispose

相关用法

注：本文由纯净天空筛选整理自nidhi1352singh大神的英文原创作品 Tensorflow.js tf.GraphModel class .dispose() Method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。