Brainstorming On Augmentations

I tried your code and nobody asked but here are my 2 cents:

  1. I don’t understand (if anybody does then please help) why you have written separate dataset classes. The dataset classes are self-sufficient on their own and are meant to be used the way they were created.
  2. You have completely neglected the pre-training phase. You are doing it in the purchase phase which is a different purpose altogether.
  3. I tried your augmentations and hyper-parameters but wasn’t able to reproduce the results. I am using the provided dataset class and pre-training phase, not the way you have done it. Maybe this could be a reason why I am not able to reproduce the results.
  4. Sorry to say this but the code is really messy.