数据集的正则化问题

VanillaVAE 模型的最后一层是 nn.Tanh()。这意味着它输出的重建图像的像素值将在 [-1, 1] 范围内。然而，问题在于 transforms.ToTensor() 将输入图像转换为 [0, 1] 范围。因此，损失是不正确的。