GitHub - afzalxo/BNN-TensorFlow: Dense BNN implementation in TensorFlow for MNIST digit classification based on https://arxiv.org/abs/1602.02830

Usage: python3 main_mnist.py

Current Acc 98.86% at 1000 Epochs.

Training took ~4 Hours on an NVIDIA Tesla V100-SXM2 16GB GPU.

GPU utiliziation using nvidia-smi:

Name	Name	Last commit message	Last commit date
Latest commit afzalxo Added visualizations using W&B Oct 13, 2019 ec70876 · Oct 13, 2019 History 23 Commits
integerized_forward_pass/bnn_inference	integerized_forward_pass/bnn_inference	Deleted main_mnist.py	Oct 6, 2019
README.md	README.md	Updated README.md	Sep 27, 2019
bnn_misc.py	bnn_misc.py	Added visualizations using W&B	Oct 13, 2019
bnn_mlp.py	bnn_mlp.py	Added visualizations using W&B	Oct 13, 2019
main_mnist.py	main_mnist.py	Added visualizations using W&B	Oct 13, 2019

Provide feedback