我正在尝试在另一个自定义层中修改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)