dichencd

Hi @stepankonev, really appreciate your great work! I tried to play with your code, but when I tried to add batch normalization layers, the training became unstable. (some values become...