@ptrblck  I have copied your augmentation code and used the same but no luck, I’m getting the same result as I used to. the only difference between my code and yours is I used a different dataset but which is almost similar to mnist but it has more labels that’s all.
you can edit the code here 1.