31 lines
790 B
31 lines
790 B
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
|