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