clean up

cec2c1da · Nicky · 6c827199 · cec2c1da · cec2c1da
Showing with 2 addition and 0 deletion

06.understand_sentiment/train_dyn_rnn.py 06.understand_sentiment/train_dyn_rnn.py +1 -0

06.understand_sentiment/train_stacked_lstm.py 06.understand_sentiment/train_stacked_lstm.py +1 -0

未找到文件。
--- 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):

--- 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):