Skip to content

Commit 655c6df

Browse files
authoredJan 9, 2018
fixed typo
1 parent 1ec90fa commit 655c6df

File tree

1 file changed

+2
-2
lines changed
  • tutorials/03-advanced/variational_auto_encoder

1 file changed

+2
-2
lines changed
 

‎tutorials/03-advanced/variational_auto_encoder/main.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def __init__(self, image_size=784, h_dim=400, z_dim=20):
3737
nn.Linear(h_dim, image_size),
3838
nn.Sigmoid())
3939

40-
def reparametrize(self, mu, log_var):
40+
def reparameterize(self, mu, log_var):
4141
""""z = mean + eps * sigma where eps is sampled from N(0, 1)."""
4242
eps = to_var(torch.randn(mu.size(0), mu.size(1)))
4343
z = mu + eps * torch.exp(log_var/2) # 2 for convert var to std
@@ -46,7 +46,7 @@ def reparametrize(self, mu, log_var):
4646
def forward(self, x):
4747
h = self.encoder(x)
4848
mu, log_var = torch.chunk(h, 2, dim=1) # mean and log variance.
49-
z = self.reparametrize(mu, log_var)
49+
z = self.reparameterize(mu, log_var)
5050
out = self.decoder(z)
5151
return out, mu, log_var
5252

0 commit comments

Comments
 (0)
Please sign in to comment.