Skip to content

Commit

Permalink
Update tf_model.py
Browse files Browse the repository at this point in the history
  • Loading branch information
npuichigo committed Oct 17, 2017
1 parent 7948004 commit 28e94ee
Showing 1 changed file with 0 additions and 11 deletions.
11 changes: 0 additions & 11 deletions src/models/tf_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,12 +76,6 @@ def __init__(self, rnn_cell, num_hidden, dnn_depth, rnn_depth, output_size,
activate_final=True,
name="mlp_input")

#self._quasi_rnns = [QuasiRNN(filter_width=2,
# num_hidden=256,
# pool_type='fo',
# name="qusi_rnn_{}".format(i))
# for i in xrange(4)]

if not self._bidirectional:
self._rnns = [
self._cell_fn(self._num_hidden)
Expand Down Expand Up @@ -136,14 +130,9 @@ def _build(self, input_sequence, input_length):
`[truncation_length, batch_size, output_size]`, and the
final state of the unrolled core,.
"""

batch_input_module = snt.BatchApply(self._input_module)
output_sequence = batch_input_module(input_sequence)

#for layer_id in xrange(2):
# quasi_rnn = self._quasi_rnns[layer_id]
# output_sequence, final_state = quasi_rnn(output_sequence, input_length)

if not self._bidirectional:
output_sequence, final_state = tf.nn.dynamic_rnn(
cell=self._core,
Expand Down

0 comments on commit 28e94ee

Please sign in to comment.