Shanmukh-C

Results 2 issues of Shanmukh-C

![image](https://user-images.githubusercontent.com/85222764/184441452-bf0e22da-0732-48b8-9325-c7d55c033b31.png) Hello we are getting this issue while running demo.py.

I tried training resnet20 on cifar10 dataset and have some surprising results could you please let me know what might be going wrong (I have just changed arch and num_classes...