Method decode_code

tok/ar_dtok/vqvae.py:92–95

(self, code_b, shape=None, channel_first=True)

        return dec

    def decode_code(self, code_b, shape=None, channel_first=True):
        quant_b = self.quantize.get_codebook_entry(code_b, shape, channel_first)
        dec = self.decode(quant_b)
        return dec

    def forward(self, input):
        quant, diff, _ = self.encode(input)