Vq Vae

Module: generative_models.models.vae.vq_vae

Source: generative_models/models/vae/vq_vae.py

Overview

Vector Quantized VAE Implementation.

Classes

VQVAE

class VQVAE

Functions

call

def __call__()

init

def __init__()

decode

def decode()

encode

def encode()

generate

def generate()

loss_fn

def loss_fn()

quantize

def quantize()

sample

def sample()

Module Statistics

• Classes: 1
• Functions: 8
• Imports: 5