I think it meant in the training stage, not inference.