diff --git a/DeepSpeech.ipynb b/DeepSpeech.ipynb index 7dbfce11..9c1216a6 100644 --- a/DeepSpeech.ipynb +++ b/DeepSpeech.ipynb @@ -528,7 +528,7 @@ " \n", " # Reshape outputs from a list of n_steps tensors each of shape [batch_size, 2*n_cell_dim]\n", " # to a single tensor of shape [n_steps*batch_size, 2*n_cell_dim]\n", - " outputs = tf.pack(outputs[0])\n", + " outputs = tf.pack(outputs)\n", " outputs = tf.reshape(outputs, [-1, 2*n_cell_dim])\n", " \n", " #Hidden layer with clipped RELU activation and dropout\n",