- nn.ConvTranspose2d(embedding_dim, nb_channels, kernel_size = 4),
- nn.ReLU(inplace = True),
- nn.ConvTranspose2d(nb_channels, nb_channels, kernel_size = 3, stride = 2), # from 4x4
- nn.ReLU(inplace = True),
- nn.ConvTranspose2d(nb_channels, nb_channels, kernel_size = 4, stride = 2), # from 9x9
- nn.ReLU(inplace = True),
- nn.ConvTranspose2d(nb_channels, nb_channels, kernel_size = 5), # from 20x20
- nn.ReLU(inplace = True),
- nn.ConvTranspose2d(nb_channels, 1, kernel_size = 5), # from 24x24
+ nn.ConvTranspose2d(embedding_dim, nb_channels, kernel_size=4),
+ nn.ReLU(inplace=True),
+ nn.ConvTranspose2d(
+ nb_channels, nb_channels, kernel_size=3, stride=2
+ ), # from 4x4
+ nn.ReLU(inplace=True),
+ nn.ConvTranspose2d(
+ nb_channels, nb_channels, kernel_size=4, stride=2
+ ), # from 9x9
+ nn.ReLU(inplace=True),
+ nn.ConvTranspose2d(nb_channels, nb_channels, kernel_size=5), # from 20x20
+ nn.ReLU(inplace=True),
+ nn.ConvTranspose2d(nb_channels, 1, kernel_size=5), # from 24x24