I don’t have any specific layers in mind, as 0.4 is quite old by now and I don’t remember exactly what has changed since then. However, it seems that you’ve already isolated it to nn.Upsample or are you not sure where the difference is coming from?