首先可以使用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)
可以看见已经发生改变。
首先可以使用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)
可以看见已经发生改变。