Backtracked so that training is measured in updates (not epochs)