You are viewing a plain text version of this content. The canonical link for it is here.
Posted to dev@mxnet.apache.org by Joern Kottmann <ko...@gmail.com> on 2017/10/12 07:09:40 UTC

Adam Update and its inputs/outputs

Hello all,

while working on the MNIST example for the Java API I noticed that the
adam_update takes four NDArrays as input, but only declares one
output. Even tough it changes three of the NDArrays.

The inputs are weight, grad, mean and var. And the output is weight,
but I had the impression it should be weight, mean and var.

Would it make sense to change that?

Jörn