Skip to content

Commit 2601c92

Browse files
committed
Add 1-layer-32-unit RNN
1 parent a9b2060 commit 2601c92

File tree

4 files changed

+2
-2
lines changed

4 files changed

+2
-2
lines changed
-3.03 MB
Binary file not shown.

dagger/model/model.index

-85 Bytes
Binary file not shown.

dagger/model/model.meta

18.2 KB
Binary file not shown.

dagger/models.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ def __init__(self, state_dim, action_cnt):
2828
# self.input: [batch_size, max_time, state_dim]
2929
self.input = tf.placeholder(tf.float32, [None, None, state_dim])
3030

31-
self.num_layers = 2
32-
self.lstm_dim = 256
31+
self.num_layers = 1
32+
self.lstm_dim = 32
3333
stacked_lstm = rnn.MultiRNNCell([rnn.BasicLSTMCell(self.lstm_dim)
3434
for _ in xrange(self.num_layers)])
3535

0 commit comments

Comments
 (0)