(self, x, return_encfea=False)
| 337 | # print(f"Unexpected Keys: {unexpected}") |
| 338 | |
| 339 | def encode(self, x, return_encfea=False): |
| 340 | h = self.encoder(x) |
| 341 | moments = self.quant_conv(h) |
| 342 | posterior = DiagonalGaussianDistribution(moments) |
| 343 | if return_encfea: |
| 344 | return posterior, moments |
| 345 | return posterior |
| 346 | |
| 347 | def encode_gt(self, x, new_encoder): |
| 348 | h = new_encoder(x) |
no test coverage detected