Views
No views yet
1import torch
2import torch.nn as nn
3
4# Define the architecture
5class DeepVAE(nn.Module):
6 def __init__(self, input_dim=15, latent_dim=4):
7 super(DeepVAE, self).__init__()
8 self.encoder = nn.Sequential(
9 nn.Linear(input_dim, 32), nn.ReLU(),
10 nn.Linear(32, 16), nn.ReLU(),
11 nn.Linear(16, latent_dim * 2)
12 )
13 self.decoder = nn.Sequential(
14 nn.Linear(latent_dim, 16), nn.ReLU(),
15 nn.Linear(16, 32), nn.ReLU(),