(latent_channels=4)
| 37 | |
| 38 | |
| 39 | def Decoder(latent_channels=4): |
| 40 | return nn.Sequential( |
| 41 | Clamp(), conv(latent_channels, 64), nn.ReLU(), |
| 42 | Block(64, 64), Block(64, 64), Block(64, 64), nn.Upsample(scale_factor=2), conv(64, 64, bias=False), |
| 43 | Block(64, 64), Block(64, 64), Block(64, 64), nn.Upsample(scale_factor=2), conv(64, 64, bias=False), |
| 44 | Block(64, 64), Block(64, 64), Block(64, 64), nn.Upsample(scale_factor=2), conv(64, 64, bias=False), |
| 45 | Block(64, 64), conv(64, 3), |
| 46 | ) |
| 47 | |
| 48 | |
| 49 | class TinyAutoencoderKL(nn.Module): |