mxnet-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From Joern Kottmann <kottm...@gmail.com>
Subject Adam Update and its inputs/outputs
Date Thu, 12 Oct 2017 07:09:40 GMT
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

Mime
View raw message