Cannot produce plausible results?

[MarkMoHR]

Hi @carpedm20 , @reiinakano , Thanks for the great work of reimplementation. But I wonder whether we can obtain plausible results with the currently released code (under mnist/simple_mnist env)? I can start up the training, but it seems that the performance is always poor even after many training steps...