Advertisement
Guest User

Untitled

a guest
Apr 21st, 2018
81
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.57 KB | None | 0 0
  1. from torch.nn import Module
  2.  
  3.  
  4. class Encoder(Module):
  5.     def __init__(self):
  6.         super().__init__()
  7.  
  8.     def forward(self):
  9.         pass
  10.  
  11.  
  12. class Decoder(Module):
  13.     def __init__(self, encoder):
  14.         super().__init__()
  15.         self.encoder = encoder
  16.  
  17.     def forward(self):
  18.         latent_space = self.encoder()
  19.         pass
  20.  
  21.  
  22. class Autoencoder(Module):
  23.     def __init__(self, Encoder, Decoder):
  24.         super().__init__()
  25.         self.decoder = Decoder()
  26.         self.encoder1 = Encoder(self.decoder)
  27.         self.encoder2 = Encoder(self.decoder)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement