Thanks @stu1130 . The implementation is not fully working. Though it can pass the unit test and can even train for a few epochs. But it will fail in the middle of the training. I will debug it later when I have more time.
[ Full content available at: https://github.com/apache/incubator-mxnet/pull/11785 ] This message was relayed via gitbox.apache.org for [email protected]
