From 156f29daed794a18b498cf0d13e40eceacae550c Mon Sep 17 00:00:00 2001 From: taixiurong Date: Thu, 17 Dec 2020 13:32:53 +0800 Subject: [PATCH] modify GPU default config(#5089) --- PaddleNLP/legacy/pretrain_language_models/BERT/README.md | 1 + .../legacy/pretrain_language_models/BERT/run_classifier.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/PaddleNLP/legacy/pretrain_language_models/BERT/README.md b/PaddleNLP/legacy/pretrain_language_models/BERT/README.md index 2f181768..b7f40a78 100644 --- a/PaddleNLP/legacy/pretrain_language_models/BERT/README.md +++ b/PaddleNLP/legacy/pretrain_language_models/BERT/README.md @@ -202,6 +202,7 @@ DATA_PATH=/path/to/xnli/data/ CKPT_PATH=/path/to/save/checkpoints/ python -u run_classifier.py --task_name ${TASK_NAME} \ + --use_cuda false \ --use_xpu true \ --do_train true \ --do_val true \ diff --git a/PaddleNLP/legacy/pretrain_language_models/BERT/run_classifier.py b/PaddleNLP/legacy/pretrain_language_models/BERT/run_classifier.py index 4669ffb3..1ead27ab 100644 --- a/PaddleNLP/legacy/pretrain_language_models/BERT/run_classifier.py +++ b/PaddleNLP/legacy/pretrain_language_models/BERT/run_classifier.py @@ -100,7 +100,7 @@ run_type_g.add_arg("profiler_path", str, './', "the profiler o run_type_g.add_arg("is_profiler", int, 0, "the profiler switch. (used for benchmark)") run_type_g.add_arg("max_iter", int, 0, "the max batch nums to train. (used for benchmark)") -run_type_g.add_arg("use_cuda", bool, False, "If set, use GPU for training.") +run_type_g.add_arg("use_cuda", bool, True, "If set, use GPU for training.") run_type_g.add_arg("use_xpu", bool, False, "If set, use XPU for training.") run_type_g.add_arg("use_fast_executor", bool, False, "If set, use fast parallel executor (in experiment).") run_type_g.add_arg("shuffle", bool, True, "") -- GitLab