unitary-nn-design

Unitary Neural Network Design

Running

Install dependencies:

apt install graphviz-dev pkg-config
pip install pygraphviz gym networkx torch==0.4.0

Install the PyTorch extension for the primary network:

cd dagnn
python setup.py install
cd ..

You may need to run with an OpenMP flag:

# get argparse help
OMP_NUM_THREADS=1 python main.py -h