提交 cec2c1da 编写于 作者: N Nicky

clean up

上级 6c827199
...@@ -24,6 +24,7 @@ CLASS_DIM = 2 ...@@ -24,6 +24,7 @@ CLASS_DIM = 2
EMB_DIM = 128 EMB_DIM = 128
BATCH_SIZE = 128 BATCH_SIZE = 128
LSTM_SIZE = 128 LSTM_SIZE = 128
USE_GPU = False
def dynamic_rnn_lstm(data, input_dim, class_dim, emb_dim, lstm_size): def dynamic_rnn_lstm(data, input_dim, class_dim, emb_dim, lstm_size):
......
...@@ -25,6 +25,7 @@ EMB_DIM = 128 ...@@ -25,6 +25,7 @@ EMB_DIM = 128
HID_DIM = 512 HID_DIM = 512
STACKED_NUM = 3 STACKED_NUM = 3
BATCH_SIZE = 128 BATCH_SIZE = 128
USE_GPU = False
def stacked_lstm_net(data, input_dim, class_dim, emb_dim, hid_dim, stacked_num): def stacked_lstm_net(data, input_dim, class_dim, emb_dim, hid_dim, stacked_num):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册