(self, x)
| 88 | |
| 89 | |
| 90 | def encode(self, x): |
| 91 | h = self.encoder(x) |
| 92 | moments = self.quant_conv(h) |
| 93 | posterior = DiagonalGaussianDistribution(moments) |
| 94 | return posterior |
| 95 | |
| 96 | def decode(self, z): |
| 97 | z = self.post_quant_conv(z) |
no test coverage detected