parent
a8342d073d
commit
9d377f0996
@ -1,42 +0,0 @@
|
|||||||
set -e
|
|
||||||
|
|
||||||
function gen_file() {
|
|
||||||
if [ ! -d "train.txt" ]; then
|
|
||||||
for ((i=1;i<=1024;i++))
|
|
||||||
do
|
|
||||||
echo "train/n09246464/n09246464_38735.jpeg 972" >> train.txt
|
|
||||||
done
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ ! -d "train.list" ]; then
|
|
||||||
echo "train.txt" > train.list
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
function train() {
|
|
||||||
cfg=$1
|
|
||||||
thread=$2
|
|
||||||
bz=$3
|
|
||||||
args="batch_size=$3"
|
|
||||||
prefix=$4
|
|
||||||
paddle train --job=time \
|
|
||||||
--config=$cfg \
|
|
||||||
--use_gpu=True \
|
|
||||||
--trainer_count=$thread \
|
|
||||||
--log_period=10 \
|
|
||||||
--test_period=100 \
|
|
||||||
--config_args=$args \
|
|
||||||
> logs/$prefix-${thread}gpu-$bz.log 2>&1
|
|
||||||
}
|
|
||||||
|
|
||||||
gen_file
|
|
||||||
if [ ! -d "logs" ]; then
|
|
||||||
mkdir logs
|
|
||||||
fi
|
|
||||||
|
|
||||||
#========multi-gpus=========#
|
|
||||||
train alexnet.py 4 512 alexnet
|
|
||||||
train alexnet.py 4 1024 alexnet
|
|
||||||
|
|
||||||
train googlenet.py 4 512 googlenet
|
|
||||||
train googlenet.py 4 1024 googlenet
|
|
@ -1,34 +0,0 @@
|
|||||||
set -e
|
|
||||||
|
|
||||||
function train() {
|
|
||||||
cfg=$1
|
|
||||||
thread=$2
|
|
||||||
args="lstm_num=${3},seq_pad=${4},hidden_size=${5},batch_size=${6}"
|
|
||||||
paddle train --job=time \
|
|
||||||
--config=$cfg \
|
|
||||||
--use_gpu=1 \
|
|
||||||
--trainer_count=$thread \
|
|
||||||
--log_period=10 \
|
|
||||||
--test_period=100 \
|
|
||||||
--num_passes=1 \
|
|
||||||
--feed_data=1 \
|
|
||||||
--config_args=$args \
|
|
||||||
>logs/rnn-pad${4}-${thread}gpu-lstm${3}-hid${5}-batch${6}.log 2>&1
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if [ ! -d "logs" ]; then
|
|
||||||
mkdir logs
|
|
||||||
fi
|
|
||||||
|
|
||||||
#-----config--gpu--lstm_num--padding--hidden_size--batch_size
|
|
||||||
#==================multi gpus=====================#
|
|
||||||
# hidden_size=256, lstm_num=2, different batch size
|
|
||||||
train rnn.py 4 2 1 256 128
|
|
||||||
train rnn.py 4 2 1 256 256
|
|
||||||
train rnn.py 4 2 1 256 512
|
|
||||||
|
|
||||||
# hidden_size=512, lstm_num=4, different batch size
|
|
||||||
train rnn.py 4 2 1 512 128
|
|
||||||
train rnn.py 4 2 1 512 256
|
|
||||||
train rnn.py 4 2 1 512 512
|
|
Loading…
Reference in new issue