- 40% of the original number of parameters of ConvNext Small
- Reshaped blocks and widths for better results on face classification
- Several additional augmentations to help generalize the model
Additionally, I tried many different methods with the scheduler. The most accurate information on the ablations is logged on wandb: https://wandb.ai/sushantj/hw2p2-ablations?workspace=user-sushantj