提交 34055f49 编写于 作者: F Francois Chollet

Quick fix for cuDNN rnns

上级 3f148e40
......@@ -40,7 +40,9 @@ class _CuDNNRNN(RNN):
for dim in self.cell.state_size]
else:
self.state_spec = InputSpec(shape=(None, self.cell.state_size))
self.constants_spec = None
self._states = None
self._num_constants = None
def _canonical_to_params(self, weights, biases):
import tensorflow as tf
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册