未验证 提交 156f29da 编写于 作者: T taixiurong 提交者: GitHub

modify GPU default config(#5089)

上级 8cd70014
...@@ -202,6 +202,7 @@ DATA_PATH=/path/to/xnli/data/ ...@@ -202,6 +202,7 @@ DATA_PATH=/path/to/xnli/data/
CKPT_PATH=/path/to/save/checkpoints/ CKPT_PATH=/path/to/save/checkpoints/
python -u run_classifier.py --task_name ${TASK_NAME} \ python -u run_classifier.py --task_name ${TASK_NAME} \
--use_cuda false \
--use_xpu true \ --use_xpu true \
--do_train true \ --do_train true \
--do_val true \ --do_val true \
......
...@@ -100,7 +100,7 @@ run_type_g.add_arg("profiler_path", str, './', "the profiler o ...@@ -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("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("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_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("use_fast_executor", bool, False, "If set, use fast parallel executor (in experiment).")
run_type_g.add_arg("shuffle", bool, True, "") run_type_g.add_arg("shuffle", bool, True, "")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册