Actual tweaks to the training algo (whoops).