You’re conflating training and inference