Hi, I had the same problem,
s_list = []
            for i in range(x1.size()[1]):
                x3 = x1[:,i,:,:].unsqueeze(1)
                x = self.up(x3)
                s_list.append(x)

self.up  is a pretrained network.
so could you give me some suggestions?