added weight histograms and graded learning rates