Revisit down_weight arg

[kels271828]

The down_weight argument was originally down_weight / (num_train - 1) instead of just down_weight. Should users take care of that part themselves, or should we put that back in? Also, what about group normalization (like with depth kernel) to make sure no dimension group has too much influence just based on number of training points?