train.sh 510 字节
Newer Older
Y
Yibing Liu 已提交
1 2
export CUDA_VISIBLE_DEVICES=0
export FLAGS_eager_delete_tensor_gb=0.0
Y
Yibing Liu 已提交
3 4
python -u ../train_and_evaluate.py --use_cuda \
                --data_path ./data/data.pkl \
Y
Yibing Liu 已提交
5 6
                --ext_eval \
                --word_emb_init ./data/word_embedding.pkl \
Y
Yibing Liu 已提交
7
                --save_path ./models \
8
                --use_pyreader \
Y
Yibing Liu 已提交
9
                --batch_size 256 \
Y
Yibing Liu 已提交
10
                --vocab_size 172130 \
Y
Yibing Liu 已提交
11
                --channel1_num 16 \
Y
Yibing Liu 已提交
12 13 14
                --emb_size 200 \
                --_EOS_ 1