Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from torch.nn import Module
- class Encoder(Module):
- def __init__(self):
- super().__init__()
- def forward(self):
- pass
- class Decoder(Module):
- def __init__(self, encoder):
- super().__init__()
- self.encoder = encoder
- def forward(self):
- latent_space = self.encoder()
- pass
- class Autoencoder(Module):
- def __init__(self, Encoder, Decoder):
- super().__init__()
- self.decoder = Decoder()
- self.encoder1 = Encoder(self.decoder)
- self.encoder2 = Encoder(self.decoder)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement