Fix cuDNN RNNs (keras-team#8244)

fchollet · web-flow · commit 9fecc5103723 · 2017-10-25T14:05:42.000-07:00
* Quick fix cudnn rnns

* Tests passing.
diff --git a/keras/layers/cudnn_recurrent.py b/keras/layers/cudnn_recurrent.py
@@ -36,10 +36,11 @@ def __init__(self,
         self.supports_masking = False
         self.input_spec = [InputSpec(ndim=3)]
         if hasattr(self.cell.state_size, '__len__'):
-            self.state_spec = [InputSpec(shape=(None, dim))
-                               for dim in self.cell.state_size]
+            state_size = self.cell.state_size
         else:
-            self.state_spec = InputSpec(shape=(None, self.cell.state_size))
+            state_size = [self.cell.state_size]
+        self.state_spec = [InputSpec(shape=(None, dim))
+                           for dim in state_size]
         self.constants_spec = None
         self._states = None
         self._num_constants = None
diff --git a/keras/layers/recurrent.py b/keras/layers/recurrent.py
@@ -454,9 +454,10 @@ def build(self, input_shape):
             # initial_state was passed in call, check compatibility
             if not [spec.shape[-1] for spec in self.state_spec] == state_size:
                 raise ValueError(
-                    'an initial_state was passed that is not compatible with'
-                    ' cell.state_size, state_spec: {}, cell.state_size:'
-                    ' {}'.format(self.state_spec, self.cell.state_size))
+                    'An initial_state was passed that is not compatible with '
+                    '`cell.state_size`. Received `state_spec`={}; '
+                    'However `cell.state_size` is '
+                    '{}'.format(self.state_spec, self.cell.state_size))
         else:
             self.state_spec = [InputSpec(shape=(None, dim))
                                for dim in state_size]
