run_CPU.sh 1.1 KB
Newer Older
Y
Yibing Liu 已提交
1 2 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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
export CPU_NUM=1

#train on ubuntu 
python -u main.py \
  --do_train True \
  --data_path ./data/ubuntu/data_small.pkl \
  --save_path ./model_files_cpu/ubuntu \
  --use_pyreader \
  --stack_num 2 \
  --vocab_size 434512 \
  --_EOS_ 28270 \
  --batch_size 32

#test on ubuntu
python -u main.py \
  --do_test True \
  --data_path ./data/ubuntu/data_small.pkl \
  --save_path ./model_files_cpu/ubuntu/step_31 \
  --model_path ./model_files_cpu/ubuntu/step_31 \
  --stack_num 2 \
  --vocab_size 434512 \
  --_EOS_ 28270 \
  --batch_size 40

#train on douban
python -u main.py \
  --do_train True \
  --data_path ./data/douban/data_small.pkl \
  --save_path ./model_files_cpu/douban \
  --use_pyreader \
  --stack_num 2 \
  --vocab_size 172130 \
  --_EOS_ 1 \
  --channel1_num 16 \
  --batch_size 32

#test on douban
python -u main.py \
  --do_test True \
  --ext_eval \
  --data_path ./data/douban/data_small.pkl \
  --save_path ./model_files_cpu/douban/step_31 \
  --model_path ./model_files_cpu/douban/step_31 \
  --stack_num 2 \
  --vocab_size 172130 \
  --_EOS_ 1 \
  --channel1_num 16 \
  --batch_size 40