如何替换 Tensorflow 中图形的现有节点？

Question

I already have a pretrained graph with me.I did some calculations on the existing weight values to get fused batchnorm weights(I have fused batchnorm and convolution weights). Using my code, I was able to save these new weight to files. Now I wanted to replace this existing weights with new weights on the graph. How this can be done? Any help would be appreciated.I am pretty new to this.

Answer 1

Use Layer.set_weights() .

See here https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer#set_weights