gen task add use_gpu (#775)

27e500be · kinghuin · GitHub · 8aeb0ed2 · 27e500be
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

demo/text_generation/run_text_gen.sh demo/text_generation/run_text_gen.sh +2 -1

未找到文件。
--- a/demo/text_generation/run_text_gen.sh
+++ b/demo/text_generation/run_text_gen.sh
@@ -9,4 +9,5 @@ python -u text_gen.py \
                   --max_seq_len 50 \
                   --learning_rate 5e-3 \
                   --cut_fraction 0.1 \
-                   --use_data_parallel True
+                   --use_data_parallel True \
+                   --use_gpu True