如何删除 Tensorflow.js 中的 model？

Question

在阅读官方文档时，我知道直接处理张量时，必须在使用后明确销毁它们以避免 memory 泄漏：

let mytensor = tf.tensor([1, 2, 3, 4]);
tf.dispose(mytensor);

但是，如何处理model ？ 只是调用mymodel.dispose()就足够了，还是我应该做其他事情？

Answer 1

model 包含层。 每层都包含可以设置的权重。 要处理这些权重，可以使用layer.dispose 。

model.layers.forEach(l => l.dispose())