From 27e500bef247bf56820bc4b19723a43e7c560dc3 Mon Sep 17 00:00:00 2001
From: kinghuin <kinghuin_chull@163.com>
Date: Mon, 27 Jul 2020 10:13:32 +0800
Subject: [PATCH] gen task add use_gpu (#775)

---
 demo/text_generation/run_text_gen.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/demo/text_generation/run_text_gen.sh b/demo/text_generation/run_text_gen.sh
index cc7a09e9..7d05f166 100644
--- 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
-- 
GitLab