search

kumar-shridhar / PyTorch-BayesianCNN

Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.
MIT License
1.42k stars 323 forks source link

Restructure image recognition #42

Closed Piyush-555 closed 4 years ago

Piyush-555 commented 4 years ago

@kumar-shridhar I have made a separate folder 'Image Recognition Restructured' temporarily. Once this PR is ready to be replaced, I'll delete the old one. Able to access any combination of (AlexNet, LeNet, 3conv3fc), (Frequentist, Bayesian), (MNIST, CIFAR10, CIFAR100) in main.py. Added tests for Layers, Models, etc. Modified .gitignore to filter out __pycache__ and .pyc files.

Image Recognition Restructured ├── experiments │ ├── extraction │ ├── figures │ └── results │ ├── cifar10 │ ├── cifar100 │ ├── mnist │ └── plots ├── layers │ ├── BBBConv.py │ ├── BBBLinear.py │ ├── misc.py ├── main.py ├── models │ ├── BayesianModels │ │ ├── Bayesian3Conv3FC.py │ │ ├── BayesianAlexNet.py │ │ ├── BayesianLeNet.py │ └── NonBayesianModels │ ├── AlexNet.py │ ├── LeNet.py │ └── ThreeConvThreeFC.py ├── README.md ├── tests.py └── utils.py