我能找到问题的答案。
更新渐变后,您必须更新权重(参数)
如果使用SGD,你会做这样的事情:
model.update(averageGradient); INDArray params = model.params(true); params.addi(averageGradient.gradient());
我希望它可以帮助其他有类似问题的人!