run.sh 582 字节
Newer Older
C
chengduoZH 已提交
1 2
#!/bin/bash

D
dangqingqing 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
set -e

function test() {
  cfg=$1
  batch_size=$2
  prefix=$3
  python $cfg --batch_size=$batch_size > logs/${prefix}-1gpu-${batch_size}.log 2>&1
}

if [ ! -d "logs" ]; then
  mkdir logs
fi

# alexnet
test alexnet.py 64 alexnet
test alexnet.py 128 alexnet
test alexnet.py 256 alexnet
test alexnet.py 512 alexnet

# googlenet
test googlenet.py 64 googlenet
test googlenet.py 128 googlenet

# smallnet 
test smallnet_mnist_cifar.py 64 smallnet
test smallnet_mnist_cifar.py 128 smallnet
test smallnet_mnist_cifar.py 256 smallnet
test smallnet_mnist_cifar.py 512 smallnet