experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make DropoutWrapper.__call__ forward the scope argument to the cell that it wraps.

Browse Source 
Change: 126222638
This commit is contained in:
A. Unique TensorFlower 2016-06-29 11:53:55 -08:00 committed by TensorFlower Gardener
parent 4e5f9a6d42
commit f2c3b2e702
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/ops/rnn_cell.py
View File

@ -660,7 +660,7 @@ class DropoutWrapper(RNNCell):
if (not isinstance(self._input_keep_prob, float) or if (not isinstance(self._input_keep_prob, float) or
self._input_keep_prob < 1): self._input_keep_prob < 1):
inputs = nn_ops.dropout(inputs, self._input_keep_prob, seed=self._seed) inputs = nn_ops.dropout(inputs, self._input_keep_prob, seed=self._seed)
output, new_state = self._cell(inputs, state) output, new_state = self._cell(inputs, state, scope)
if (not isinstance(self._output_keep_prob, float) or if (not isinstance(self._output_keep_prob, float) or
self._output_keep_prob < 1): self._output_keep_prob < 1):
output = nn_ops.dropout(output, self._output_keep_prob, seed=self._seed) output = nn_ops.dropout(output, self._output_keep_prob, seed=self._seed)