train.sh 449 字节
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
                --word_emb_init ./data/word_embedding.pkl \
Y
Yibing Liu 已提交
6
                --save_path ./models \
7
                --use_pyreader \
Y
Yibing Liu 已提交
8
                --batch_size 256 \
Y
Yibing Liu 已提交
9 10 11 12
                --vocab_size 434512 \
                --emb_size 200 \
                --_EOS_ 28270