【发布时间】:2020-11-18 19:41:54
【问题描述】:
我已经有了一个预训练的图表。我对现有的权重值进行了一些计算以获得融合的 batchnorm 权重(我已经融合了 batchnorm 和卷积权重)。使用我的代码,我能够将这些新权重保存到文件中。现在我想用图表上的新权重替换现有的权重。 如何做到这一点? 任何帮助将不胜感激。我对此很陌生。
【问题讨论】:
标签: python tensorflow graph google-colaboratory
我已经有了一个预训练的图表。我对现有的权重值进行了一些计算以获得融合的 batchnorm 权重(我已经融合了 batchnorm 和卷积权重)。使用我的代码,我能够将这些新权重保存到文件中。现在我想用图表上的新权重替换现有的权重。 如何做到这一点? 任何帮助将不胜感激。我对此很陌生。
【问题讨论】:
标签: python tensorflow graph google-colaboratory
使用Layer.set_weights()。
请看这里https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer#set_weights
【讨论】: