model.py 文件源码

def forward(self, x):
        out_1 = F.leaky_relu(self.conv1(x), 0.05)      # (?, 64, 16, 16)
        out_2 = F.leaky_relu(self.conv2(out_1), 0.05)    # (?, 128, 8, 8)

        out_3 = F.leaky_relu(self.conv3(out_2), 0.05)    # ( " )
        out_4 = F.leaky_relu(self.conv4(out_3), 0.05)    # ( " )

        out_5 = F.leaky_relu(self.deconv1(out_4), 0.05)  # (?, 64, 16, 16)
        out = F.tanh(self.deconv2(out_5))              # (?, 1, 32, 32)

        return out