add mini-batch training, weights look much better