Solved by vabh in post #2 
              
              
                I think a simple way to define a module to use the interpolate function. Then you can use it in your sequential container same as you are using nn.Upsample currently. 
Something along these lines: 
class Upsample(nn.Module):
    def __init__(self,  scale_factor):
        super(Upsample, self).__init…