fix grad in generative model
test on MNIST
mend
check in small mnist data