DL4J。在做了model.upgrade（Gradient）后权重更新了吗？

Question

我正在使用MultiLayerNetwork。该网络经过培训，结果如预期。 作为实施Federated Learning的努力的一部分，我需要使用来自外部组件的一组新渐变来升级此网络的权重。

我的问题是，在DL4J中，一旦执行model.update(newGradients)，权重是否也会更新，还是我需要执行其他内容？

由于

Answer 1

我能找到问题的答案。

更新渐变后，您必须更新权重（参数）

如果使用SGD，您可以这样做：

model.update(averageGradient);
INDArray params = model.params(true);
params.addi(averageGradient.gradient());

我希望它可以帮助其他有类似问题的人！