(self, x)
| 571 | self.embed_dim = embed_dim |
| 572 | |
| 573 | def encode(self, x): |
| 574 | h = self.encoder(x) |
| 575 | moments = self.quant_conv(h) |
| 576 | posterior = DiagonalGaussianDistribution(moments) |
| 577 | return posterior |
| 578 | |
| 579 | def decode(self, z): |
| 580 | z = self.post_quant_conv(z) |
no test coverage detected