Beta Vae

Module: generative_models.models.vae.beta_vae

Source: generative_models/models/vae/beta_vae.py

Overview

Beta Variational Autoencoder implementation.

Classes

BetaVAE

class BetaVAE

BetaVAEWithCapacity

class BetaVAEWithCapacity

Functions

init

def __init__()

init

def __init__()

loss_fn

def loss_fn()

loss_fn

def loss_fn()

Module Statistics

• Classes: 2
• Functions: 4
• Imports: 4