CS772 Course Project

---

• This is our code for the CS772 project based on the paper.
• The ipynb notebook contains the result and analysis done on different CNN models, it also contains comments with brief explanation of what the paritcular code segment does.
• train.py and test_and_eval.py are to train and test the models.
• File named loss_functions contains code for different loss functions we trained the models on.
• cnn.py contains the architecture of the CNN and ABNN models we used
• bnl.py cotains Bayesian Normalisation Layer
• cifar10.py contains the preprocessing part of the code.