deep-learning-with-python-notebooks icon indicating copy to clipboard operation
deep-learning-with-python-notebooks copied to clipboard

Published 20 hours ago • verified publisher icon fchollet

The source code will cause IndexError

Open XZLeo opened this issue 4 years ago • 0 comments

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.

XZLeo avatar Jul 05 '20 06:07 XZLeo