Add Back Augmentations

[leonardoaraujosantos]

References

• Pytorch Issue Message
• Pytorch Discussion
• Pytorch Discussion 2

[leonardoaraujosantos]

I've added back some augmentations, Including the swap (reference, test) which is confirmed that does NOT help on the current architecture, also unfreezing the Resnet branches gave 12%+ accuracy boost on the validation set.

### Labels * Orange: Baseline with Resnets frozen * Magenta: Augmentations with Resnet unfrozen * Green: Swap with Augmentations and Resnet unfrozen

TODO: Still add the flip on: (Reference, Test, and label) with same probability