如何在keras自定义层中更新参数（在每个时期基于该时期的输入状态？）？]

Question

我有一个keras顺序模型，并且那里有一个自定义示例层，可以说它的名字是'LayerX'。现在在“ LayerX”中，我有一个参数“ lambda”，我想用一个值进行初始化，例如lambda = 10，

现在，在训练时，在每个时期，我都会在“ LayerX”层的“ call”方法中接受输入，并且我基于那个时期的输入来计算值“ valX”，我想在每个时期使用此值'valX'更新参数'lamba'。假设在每个时期，lambda = lambda + valX。

我是定制角膜层的新手。谁能帮助我了解如何做？

我有一个keras顺序模型，并且那里有一个自定义示例层，可以说它的名字是'LayerX'。现在在“ LayerX”中，我有一个参数“ lambda”，我想用一个值初始化...

Answer 1

这里是一个示例，我在每个时期之后都提取了梯度。您可以对model.fit循环进行更改，以对图层进行自定义更改。