(self, x)
| 62 | nn.Tanh() # if img is -1 to 1 |
| 63 | ) |
| 64 | def forward(self, x): |
| 65 | encode = self.encoder(x) |
| 66 | decode = self.decoder(encode) |
| 67 | return encode, decode |
| 68 | |
| 69 | class autoencoder_vgg2(nn.Module): # psnr 25.41 |
| 70 | ''' add skip connections ''' |
nothing calls this directly
no outgoing calls
no test coverage detected