mtcrawshaw/growing-cnns

PyTorch implementation of a training scheme for convolutional neural networks called "Growing CNNs"

growing-cnns

PyTorch implementation of a training scheme for convolutional neural networks called "Growing CNNs"

Install

This project requires Python 3.x and the following Python libraries:

• numpy
• pandas
• matplotlib
• PyTorch

Usage

To train a Growing CNN on the CIFAR-10 dataset, from the root of the project run:

python3 main.py example settings/exampleGrowingSettings.json

Contact

If you are interested in contacting the author, please use the following email: michael.crawshaw.4@gmail.com