All notebook files can be run on google colab:
CNN_withSpectrumNorm:
https://colab.research.google.com/drive/1JRWGK8bz5FaDzDR2u9siytgeg2s0gBSX?usp=sharing
CNN_withBN:
https://colab.research.google.com/drive/19WKtaJWpKGfxEnnCqZZNXhT261W4MCgH?usp=sharing
LinearWithBN:
https://colab.research.google.com/drive/1vgf-EnpW4zQEtUIxh8YRj2e38aTTCABM?usp=sharing
LinearWithOutBN:
https://colab.research.google.com/drive/1iRfEU8s56iaeLtndKpP1f5FQ-3pZRkEr?usp=sharing
In all colab notebooks or jupyter notebooks just hit and run all to replicate the results.
If for some reason an error happend, which didn't happen when i did run them all, try to
remove the assert statement that caused the error and the code will continue to work just fine