Solved by nunenuh in post #7 
              
              
                You can use this to make shape of your picture bigger, scale_factor=2 mean you scale it twice. 
 nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True)