(latent_channels=4)
| 27 | |
| 28 | |
| 29 | def Encoder(latent_channels=4): |
| 30 | return nn.Sequential( |
| 31 | conv(3, 64), Block(64, 64), |
| 32 | conv(64, 64, stride=2, bias=False), Block(64, 64), Block(64, 64), Block(64, 64), |
| 33 | conv(64, 64, stride=2, bias=False), Block(64, 64), Block(64, 64), Block(64, 64), |
| 34 | conv(64, 64, stride=2, bias=False), Block(64, 64), Block(64, 64), Block(64, 64), |
| 35 | conv(64, latent_channels), |
| 36 | ) |
| 37 | |
| 38 | |
| 39 | def Decoder(latent_channels=4): |