我正在使用PyTorch中实现的卷积自动编码器进行图像重建任务。我使用的模型如下：

class Model_CAE(nn.Module):
    def __init__(self):
        super(Model_CAE, self).__init__()
        self.act = nn.Hardtanh()
    
        # ENCODER
        self.e1 = nn.Conv2d(3, 16, kernel_size=5, stride=2, padding=0)
        self.e2 = nn.Conv2d(16, 32, kernel_size=2, stride=2, padding=2)
        self.e3 = nn.Conv2d(32, 64, kernel_size=2, stride=2, padding=0)

        # DECODER
        self.d1 = nn.ConvTranspose2d(64, 32, kernel_size=2, stride=2, padding=0, output_padding=0)
        self.d2 = nn.ConvTranspose2d(32, 16, kernel_size=2, stride=2, padding=2, output_padding=0)
        self.d3 = nn.ConvTranspose2d(16, 3, kernel_size=5, stride=2, padding=0, output_padding=1)

    def forward(self, x):
        #SENZA LINEAR
        e1 = self.act(self.e1(x))
        e2 = self.act(self.e2(e1))
        e3 = self.act(self.e3(e2))
   
        d1 = self.act(self.d1(e3))
        d2 = self.act(self.d2(d1))
        out = self.act(self.d3(d2))
    
        return out

    def name(self):
        return "CAE"

问题是，在网络的输出上出现了一种网格，如下面的示例所示：

我试图改变激活功能，但没有任何变化。我能怎么办？