Context only needed during training.