如何在Keras中修改另一层的权重?

时间:2019-09-14 19:13:47

标签: tensorflow keras

我正在尝试在另一个自定义层中修改keras层(例如Conv2D)的权重。

我试图将图层(权重将被修改)传递到自定义图层,以便可以在自定义图层中使用get_weights,set_weights方法。但是,事实证明我无法将图层或模型作为参数传递给自定义图层。

例如,我想在下面的自定义层中使用模型或特定层。为此,我需要将模型或特定层传递给ExampleFunc。请注意,我无法在_ init _。

中传递模型或图层。
    class exampleLayer(tf.keras.layers.Layer):
        def __init__(self, indexofOtherLayer):
            super(exampleLayer, self).__init__()
            self.indL = indexofOtherLayer

        def call(self, inputs, **kwargs):
            return ExampleFunc(MODEL,inputs,self.indL)

0 个答案:

没有答案