Search code examples
pythonscalamxnet

MXNet - what is python equivalent of getting scala's mxnet networkExecutor.gradDict("data")

Trying to understand some Scala's code on network training in MXNet.

I believe you can access gradient on the executor in Scala by calling networkExecutor.gradDict("data"), what would be equivalent of it in Python MXNet?

Thanks!

Solution

  • How about grad_dict in executor? It returns a dictionary representation of the gradient arrays.