From cec2c1da543d93754a575c6b28f718c2d0c50a02 Mon Sep 17 00:00:00 2001
From: Nicky <nicky@baidu.com>
Date: Wed, 13 Jun 2018 15:07:25 -0700
Subject: [PATCH] clean up

---
 06.understand_sentiment/train_dyn_rnn.py      | 1 +
 06.understand_sentiment/train_stacked_lstm.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/06.understand_sentiment/train_dyn_rnn.py b/06.understand_sentiment/train_dyn_rnn.py
index e139d67..f548a27 100644
--- a/06.understand_sentiment/train_dyn_rnn.py
+++ b/06.understand_sentiment/train_dyn_rnn.py
@@ -24,6 +24,7 @@ CLASS_DIM = 2
 EMB_DIM = 128
 BATCH_SIZE = 128
 LSTM_SIZE = 128
+USE_GPU = False
 
 
 def dynamic_rnn_lstm(data, input_dim, class_dim, emb_dim, lstm_size):
diff --git a/06.understand_sentiment/train_stacked_lstm.py b/06.understand_sentiment/train_stacked_lstm.py
index 0a3b201..104215d 100644
--- a/06.understand_sentiment/train_stacked_lstm.py
+++ b/06.understand_sentiment/train_stacked_lstm.py
@@ -25,6 +25,7 @@ EMB_DIM = 128
 HID_DIM = 512
 STACKED_NUM = 3
 BATCH_SIZE = 128
+USE_GPU = False
 
 
 def stacked_lstm_net(data, input_dim, class_dim, emb_dim, hid_dim, stacked_num):
-- 
GitLab