In the notebook, Block 5: Prepare Data for Model Training, it says to use a ImageData Generator for the image augmentation otherwise it takes forever. I can see in the cell that it does yet in the documentation it says it's deprecated. Any suggestions on what to use? Also, it seems like it is taking a very long time to perform the augmentation on the training data at this point.