Compute histogram of the weights (to visualize effect of regularization)