Be fair, don't use test data for the training. Don't use mask during training