improved computation of next training update