In the "Get to know Create ML Components" session around the 14m mark, the presenter mentioned that the augmentation applied is only used during training, not validation. Is that really true, given that it was just applied using flatMap() to the combined dataset in the code shown? It is not what I would expect based on reading the code.
It applies for training and validation data, but only at training time, not when doing predictions.