def forward(self, x):
    x = self.upsample(x)
    x = self.conv1(x)
    x = self.norm1(x)
    x = self.relu1(x)
    return x
def forward(self, x):
    x = self.upsample(x)
    x = self.conv1(x)
    x = self.norm1(x)
    x = self.relu1(x)
    return x