fixed training bug
milner bug fix