您好)是否有一种最佳方法可以将经过训练的权重从一个模型复制到 Tensorflow.js 中的另一个模型?或者如何将模型转换为字符串? 我需要从一名工作人员复制经过训练的模型权重并将其发送给另一名工作人员。
最佳答案
您可以使用 tf.layers.Layer.getWeights
获取特定层的权重。这将返回整个指定层的权重张量。您可以在另一个模型上使用 setWeights
设置这些权重。
要指定特定层,您可以使用getLayer
函数。下面是获取特定图层的权重并打印它们的示例。
tf.loadLayersModel('https://d1zv2aa70wpiur.cloudfront.net/tfjs_quant_nsfw_mobilenet/model.json')
.then(myModel => {
myModel.summary() // Lists layers
const chosenLayer = myModel.getLayer('Conv_1')
chosenLayer.getWeights()[0].print()
})
关于javascript - tensorflow.js 将权重从一个模型复制到另一个模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/68448315/