提交 4cbc4d76 编写于 作者: Z Zeyu Chen

change default value from use_cuda from False to True

上级 3f746769
...@@ -10,7 +10,7 @@ CKPT_DIR="./ckpt_${DATASET}" ...@@ -10,7 +10,7 @@ CKPT_DIR="./ckpt_${DATASET}"
python -u text_classifier.py \ python -u text_classifier.py \
--batch_size=24 \ --batch_size=24 \
--use_gpu=False \ --use_gpu=True \
--dataset=${DATASET} \ --dataset=${DATASET} \
--checkpoint_dir=${CKPT_DIR} \ --checkpoint_dir=${CKPT_DIR} \
--learning_rate=5e-5 \ --learning_rate=5e-5 \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册