This repo contains all the variants tested for Transforming Convolutional Neural Networks for Model Parallelism, the most interesting of which are reported on in said document.
Also contains code for collecting data.
- Clone the repo
git clone [email protected]:FelixBrakel/HPCNas.git - Create the conda environment
conda env create -f conda/environment.yaml - Activate the conda environment
conda activate parnassia