diff --git a/PaddleNLP/language_representations_kit/ELMo/run.sh b/PaddleNLP/language_representations_kit/ELMo/run.sh index b83c34fa3e55fc9c8ef2340df5ea7c06ad45a088..a14235135097061929417ad1a2548033541e46e7 100755 --- a/PaddleNLP/language_representations_kit/ELMo/run.sh +++ b/PaddleNLP/language_representations_kit/ELMo/run.sh @@ -1,4 +1,7 @@ +#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_*' \