未验证 提交 97c11d15 编写于 作者: J JesseyXujin 提交者: GitHub

add export cpu_nums for supporting cpu environment (#3302)

* add export cpu_nums for supporting cpu environment

* add export cpu_nums for supporting cpu environment
上级 7b7fba1c
#if use gpu, you could specify gpu devices
export CUDA_VISIBLE_DEVICES=0
#if use cpu, you could specify cpu nums
export CPU_NUM=10
python train.py \
--train_path='data/train/sentence_file_*' \
--test_path='data/dev/sentence_file_*' \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册