(self, x)
| 266 | self.embed_dim = embed_dim |
| 267 | |
| 268 | def encode(self, x): |
| 269 | h = self.encoder(x) |
| 270 | h = self.quant_conv(h) |
| 271 | return h |
| 272 | |
| 273 | def decode(self, h, force_not_quantize=False): |
| 274 | # also go through quantization layer |
nothing calls this directly
no outgoing calls
no test coverage detected