I have this model which has an encoder-decoder structure. I want to change the “functional.interpolate” layer (which does upsampling layer) to “functional.max_unpool2d”. But as you can see in the following code in the “colab link” https://colab.research.google.com/drive/1UHjW5tEEG-ZwIEIlPzVuZSgvGnPuKVMI?usp=sharing 1, the max pooling is inside a Sequential, and I couldn’t get the indices of the tensor of max pooling layer. Are there any ideas on how to change the “functional.interpolate” layer to “functional.max_unpool2d” without changing a lot in the code. Really appreciate any help. Thank you so much in advance.