Issues with Data Augmentation

[rachyuthan]

Hello,

I included code for the data augmentation to flip both the vertical and horizontal axes but when I set the augmentation to True and then run it in the visualization I do not see that a full representation on the polar plot. Images for further context:

I want to make sure that I have this correct before running the training loop so should I be seeing the results of the data augmentation already or is it only visible once the training loop has been executed.

Thanks

[yilun-wu]

Perhaps you are plotting the distribution the train_dataset. Note the augmentation is performed in the dataloader. The dataloader takes the data from train_dataset (raw unaugmented) and then augments it and return the augmented version in the training loop to feed it to the model. If you plot the distribution within the training loop, you should see a difference.