diff --git a/06.understand_sentiment/train_dyn_rnn.py b/06.understand_sentiment/train_dyn_rnn.py index e139d670f96958ecec794ee32a252d08e8f5d552..f548a27f6fb41acc0888aa61967c47a4a110fa51 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 0a3b201579b17cacb031e55e970a1ca5188516b6..104215d7d9c212cef3991442ebd6a2ab5bd03d93 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):