Skip to content

Running AlexNet (Old)

Charles Siegel edited this page Mar 28, 2017 · 1 revision

Running AlexNet

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.

Clone this wiki locally