Refactor weight annealing to an easily-switchable form.