search

fchollet / deep-learning-with-python-notebooks

Jupyter notebooks for the code samples of the book "Deep Learning with Python"
MIT License
18.61k stars 8.63k forks source link

The source code will cause IndexError #142

Open XZLeo opened 4 years ago

XZLeo commented 4 years ago

The shape of conv_layer_output_value is (1, 14, 14, 512), and the shape of pooled_grads_value is (512, ).
So it will cause IndexError when multiple conv_layer_output_value[:, :, i] with pooled_grads_value[i] in the source code. Thus I change it to conv_layer_output_value[:, :, :, i] to align the dimension.