Basic MNIST Example

pip install -r requirements.txt
python main.py
# CUDA_VISIBLE_DEVICES=2 python main.py  # to specify GPU id to ex. 2

Name		Name	Last commit message	Last commit date
Latest commit History 59 Commits
Makefile		Makefile
README.md		README.md
binary_SGD.py		binary_SGD.py
binary_layers.py		binary_layers.py
binary_net.py		binary_net.py
calc_divergence.py		calc_divergence.py
cuda_test.py		cuda_test.py
layers.py		layers.py
learn_addition.py		learn_addition.py
main.py		main.py
popc_cuda.py		popc_cuda.py
quantized_logistic.py		quantized_logistic.py
requirements.txt		requirements.txt
split_backprop_test.py		split_backprop_test.py

Provide feedback