in line 45: p3 = F.interpolate(p4, size=[p4.size(2), p4.size(3)]) + p4 should change to: p4 = F.interpolate(p5, size=[p4.size(2), p4.size(3)]) + p4