(self, x)
| 262 | return y, m, logs, y_mask |
| 263 | |
| 264 | def extract_latent(self, x): |
| 265 | x = self.ssl_proj(x) |
| 266 | quantized, codes, commit_loss, quantized_list = self.quantizer(x) |
| 267 | return codes.transpose(0, 1) |
| 268 | |
| 269 | def decode_latent(self, codes, y_mask, refer, refer_mask, ge): |
| 270 | quantized = self.quantizer.decode(codes) |
no outgoing calls
no test coverage detected