change default value from use_cuda from False to True

4cbc4d76 · Zeyu Chen · 3f746769 · 4cbc4d76
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

demo/text-classification/run_classifier.sh demo/text-classification/run_classifier.sh +1 -1

未找到文件。
--- a/demo/text-classification/run_classifier.sh
+++ b/demo/text-classification/run_classifier.sh
@@ -10,7 +10,7 @@ CKPT_DIR="./ckpt_${DATASET}"

 python -u text_classifier.py \
                   --batch_size=24 \
-                   --use_gpu=False \
+                   --use_gpu=True \
                   --dataset=${DATASET} \
                   --checkpoint_dir=${CKPT_DIR} \
                   --learning_rate=5e-5 \