MXNet C++ Package Examples

Building C++ examples

The examples in this folder demonstrate the training workflow. The inference workflow related examples can be found in inference folder. Please build the MXNet C++ Package as explained in the README File. The examples in this folder are built while building the MXNet library and cpp-package from source. You can get the executable files by just copying them from mxnet/build/cpp-package/example

The examples that are built to be run on GPU may not work on the non-GPU machines.

Examples demonstrating training workflow

This directory contains following examples. In order to run the examples, ensure that the path to the MXNet shared library is added to the OS specific environment variable viz. LD_LIBRARY_PATH for Linux, Mac and Ubuntu OS and PATH for Windows OS. For example export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/home/ubuntu/mxnet/build on ubuntu using gpu.

alexnet.cpp

The example implements the C++ version of AlexNet. The networks trains on MNIST data. The number of epochs can be specified as a command line argument. For example to train with 10 epochs use the following:

build/alexnet 10

googlenet.cpp

The code implements a GoogLeNet/Inception network using the C++ API. The example uses MNIST data to train the network. By default, the example trains the model for 100 epochs. The number of epochs can also be specified in the command line. For example, to train the model for 10 epochs use the following:

build/googlenet 10

mlp.cpp

The code implements a multilayer perceptron from scratch. The example creates its own dummy data to train the model. The example does not require command line parameters. It trains the model for 20,000 epochs. To run the example use the following command:

build/mlp

mlp_cpu.cpp

The code implements a multilayer perceptron to train the MNIST data. The code demonstrates the use of “SimpleBind” C++ API and MNISTIter. The example is designed to work on CPU. The example does not require command line parameters. To run the example use the following command:

build/mlp_cpu

mlp_gpu.cpp

The code implements a multilayer perceptron to train the MNIST data. The code demonstrates the use of the “SimpleBind” C++ API and MNISTIter. The example is designed to work on GPU. The example does not require command line arguments. To run the example execute following command:

build/mlp_gpu

mlp_csv.cpp

The code implements a multilayer perceptron to train the MNIST data. The code demonstrates the use of the “SimpleBind” C++ API and CSVIter. The CSVIter can iterate data that is in CSV format. The example can be run on CPU or GPU. The example usage is as follows:

build/mlp_csv --train data/mnist_data/mnist_train.csv --test data/mnist_data/mnist_test.csv --epochs 10 --batch_size 100 --hidden_units "128 64 64" --gpu

To get the mnist_training_set.csv and mnist_test_set.csv please run the following command:

# in mxnet/cpp-package/example directory
python mnist_to_csv.py ./data/mnist_data/train-images-idx3-ubyte ./data/mnist_data/train-labels-idx1-ubyte ./data/mnist_data/mnist_train.csv 60000
python mnist_to_csv.py ./data/mnist_data/t10k-images-idx3-ubyte ./data/mnist_data/t10k-labels-idx1-ubyte ./data/mnist_data/mnist_test.csv 10000

resnet.cpp

The code implements a resnet model using the C++ API. The model is used to train MNIST data. The number of epochs for training the model can be specified on the command line. By default, model is trained for 100 epochs. For example, to train with 10 epochs use the following command:

build/resnet 10

lenet.cpp

The code implements a lenet model using the C++ API. It uses MNIST training data in CSV format to train the network. The example does not use built-in CSVIter to read the data from CSV file. The number of epochs can be specified on the command line. By default, the mode is trained for 100,000 epochs. For example, to train with 10 epochs use the following command:

build/lenet 10

lenet_with_mxdataiter.cpp

The code implements a lenet model using the C++ API. It uses MNIST training data to train the network. The example uses built-in MNISTIter to read the data. The number of epochs can be specified on the command line. By default, the mode is trained for 100 epochs. For example, to train with 10 epochs use the following command:

build/lenet_with_mxdataiter 10

In addition, there is run_lenet_with_mxdataiter.sh that downloads the mnist data and run lenet_with_mxdataiter example.

inception_bn.cpp

The code implements an Inception network using the C++ API with batch normalization. The example uses MNIST data to train the network. The model trains for 100 epochs. The example can be run by executing the following command:

build/inception_bn