Hi,
The problem is that you’re using MaxPool2d with return_indices=True.
This means that it will return a tuple with both the value and the indices.
This is not handled by the convolution layer that follows.