Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Running AlexNet (Old)
Clone this wiki locally
MaTEx provides a script -- src/deeplearning/mpi-tensorflow_alexnet.py -- that will train AlexNet using multiple CPU, GPU compute nodes. It takes two parameters --train_data and --test_data with the arguments a pair of PNetCDF files containing the ILSVRC2012 training and validation sets.