set -e function test() { cfg=$1 batch=$2 prefix=$3 sed -i "/input: \"data\"/{n;s/^input_dim.*/input_dim: $batch/g}" $cfg sed -i "/input: \"label\"/{n;s/^input_dim.*/input_dim: $batch/g}" $cfg caffe time --model=$cfg --iterations=50 --gpu 0 > logs/$prefix-1gpu-batch${batch}.log 2>&1 } if [ ! -d "logs" ]; then mkdir logs fi # alexnet test alexnet.prototxt 64 alexnet test alexnet.prototxt 128 alexnet test alexnet.prototxt 256 alexnet test alexnet.prototxt 512 alexnet # googlenet test googlenet.prototxt 64 googlenet test googlenet.prototxt 128 googlenet # small net test smallnet_mnist_cifar.prototxt 64 smallnet test smallnet_mnist_cifar.prototxt 128 smallnet test smallnet_mnist_cifar.prototxt 256 smallnet test smallnet_mnist_cifar.prototxt 512 smallnet
