please do not overwrite scale_ (see more below). For this context you can use just b_weight.