white-box-nn

The official PyTorch implementation of paper Towards White Box Deep Learning.

Recreate env

To build environment run the following commands. It's recommended to read the env_setup.sh script first. In particular PyTorch/CUDA installation should be adjusted for your machine (however CPU-only PyTorch installation will suffice).

conda create --name wbnn pip
conda activate wbnn
bash env_setup.sh

To recreate the experiments run the provided jupyter notebooks with the adequate kernel (wbnn conda env).