diff --git a/examples/Advanced-Tutorials/autoencoders/mnist-vae.py b/examples/Advanced-Tutorials/autoencoders/mnist-vae.py index 0bfad6f8..acea0309 100644 --- a/examples/Advanced-Tutorials/autoencoders/mnist-vae.py +++ b/examples/Advanced-Tutorials/autoencoders/mnist-vae.py @@ -15,13 +15,13 @@ def __init__(self, d): super().__init__() self.d = d self.encoder = nn.Sequential( - nn.Linear(784, self.d ** 2), nn.ReLU(), nn.Linear(self.d ** 2, self.d * 2) + nn.Linear(784, self.d**2), nn.ReLU(), nn.Linear(self.d**2, self.d * 2) ) self.decoder = nn.Sequential( - nn.Linear(self.d, self.d ** 2), + nn.Linear(self.d, self.d**2), nn.ReLU(), - nn.Linear(self.d ** 2, 784), + nn.Linear(self.d**2, 784), nn.Sigmoid(), )