human
Give me a the Python code you requested given the following:
I am working on the file "Sessions/Session14/Day3/IntroductionToVariationalAutoencoders.ipynb".
The first task is:
STEP 2: Modify the decoder network.
We will leave our decoder network as it is :)
Can you write Python code for it?
assistant
#Here is the Python code you requested.

class VAEDecoder(nn.Module):

    def __init__(self, seed=620):
        super(VAEDecoder, self).__init__()
        self.seed = torch.manual_seed(seed)
        self.fc1 = nn.Linear(LATENT_SIZE,50)
        self.fc2 = nn.Linear(50,INPUT_SIZE)

    def forward(self, z):
        z = torch.nn.LeakyReLU()(self.fc1(z))
        x = self.fc2(z)
        return x