Tensorflow 中如何修改模型的权重?

首先可以使用get_weights() API获取权值。

t=watermark_model.layers[1].get_weights()

修改t(这个变量的类型和shape大家就自己看啦)

t[0][0,0,0,0]=0

使用 set_weights(weight) API 设置权重。

watermark_model.layers[1].set_weights(t)

可以看见已经发生改变。
Tensorflow 中如何修改模型的权重?_第1张图片

你可能感兴趣的:(深度学习,tensorflow)