Skip to content

Instantly share code, notes, and snippets.

@dvgodoy

dvgodoy/cvae_decoder.py

Created April 30, 2022 12:36
Show Gist options
  • Save dvgodoy/c8415eff06c6cb907f4f5c0546a93573 to your computer and use it in GitHub Desktop.
Save dvgodoy/c8415eff06c6cb907f4f5c0546a93573 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cvae_decoder.py
decoder_cnn = nn.Sequential(
# z_size -> (n_filters*2)*7*7
nn.Linear(z_size, (n_filters*2)*int(img_size/4)**2),
# (n_filters*2)*7*7 -> (n_filters*2)@7x7
nn.Unflatten(1, (n_filters*2, int(img_size/4), int(img_size/4))),
# (n_filters*2)@7x7 -> (n_filters*2)@7x7
nn.ConvTranspose2d(n_filters*2, n_filters*2, kernel_size=3, stride=1, padding=1, output_padding=0),
nn.LeakyReLU(),
# (n_filters*2)@7x7 -> (n_filters*2)@14x14
nn.ConvTranspose2d(n_filters*2, n_filters*2, kernel_size=3, stride=2, padding=1, output_padding=1),
nn.LeakyReLU(),
# (n_filters*2)@15x15 -> n_filters@28x28
nn.ConvTranspose2d(n_filters*2, n_filters, kernel_size=3, stride=2, padding=1, output_padding=1),
nn.LeakyReLU(),
# n_filters@28x28 -> in_channels@28x28
nn.ConvTranspose2d(n_filters, in_channels, kernel_size=3, stride=1, padding=1, output_padding=0),
nn.Sigmoid(),
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment