Add script to speedtest imagenet (currently broken as FilterNet ignores

Caffe::phase()).

diff --git a/examples/imagenet/speedtest_imagenet.sh b/examples/imagenet/speedtest_imagenet.sh
new file mode 100755 (executable)
index 0000000..c790d83
--- /dev/null
+++ b/examples/imagenet/speedtest_imagenet.sh
@@ -0,0 +1,22 @@
+#!/usr/bin/env sh
+
+TOOLS=../../build/tools
+
+if [ -z "$1" ]; then
+  echo "Using CPU! To run GPU speedtest, use:"
+  echo "    ./speedtest_imagenet.sh <device ID>"
+  echo "(Try ./speedtest_imagenet.sh 0 if you have just one GPU.)"
+  sleep 3  # Let the user read
+  WITH_GPU=false
+  DEVICE_ID=0
+else
+  WITH_GPU=true
+  DEVICE_ID=$1
+fi
+
+GLOG_logtostderr=1 $TOOLS/caffe.bin speedtest \
+  --net_proto_file=imagenet_train_val.prototxt \
+  --speedtest_with_gpu=${WITH_GPU} \
+  --device_id=${DEVICE_ID}
+
+echo "Done."