Breaking News: Grepper is joining You.com. Read the official announcement!

sample convolution autoencoder

josh.ipynb answered on July 24, 2023 Popularity 2/10 Helpfulness 1/10

answer sample convolution autoencoder

sample convolution autoencoder

Comment

class Autoencoder(nn.Module):
    def __init__(self, encoding_dim):
        super(Autoencoder, self).__init__()

        ## encoder ##
        self.encoder = nn.Sequential(
            nn.Conv2d(1, 3, 3, padding=1),
            nn.ReLU(),
            nn.MaxPool2d(2, 2)
        )

        ## decoder ##
        self.decoder = nn.Sequential(
            # Undo the Max Pooling
            nn.ConvTranspose2d(3, 1, 2, stride=2),
            nn.Sigmoid()
        )

        self.auto_encoder = nn.Sequential(
            self.encoder,
            self.decoder
        )

    def forward(self, x):
        # define feedforward behavior 
        # and scale the *output* layer with a sigmoid activation function

        return self.auto_encoder(x)

Popularity 2/10 Helpfulness 1/10 Language python

Source: learn.udacity.com

Tags: autoencoder convolution python

Link to this answer
Share Copy Link