Ebm

Module: generative_models.models.energy.ebm

Source: generative_models/models/energy/ebm.py

Overview

Energy-Based Model implementations.

This module provides concrete implementations of energy-based models for different data types and use cases.

Classes

DeepCNNEnergyFunction

class DeepCNNEnergyFunction

DeepEBM

class DeepEBM

EBM

class EBM

EnergyBlock

class EnergyBlock

Functions

call

def __call__()

call

def __call__()

init

def __init__()

init

def __init__()

init

def __init__()

init

def __init__()

create_cifar_ebm

def create_cifar_ebm()

create_mnist_ebm

def create_mnist_ebm()

create_simple_ebm

def create_simple_ebm()

get_config

def get_config()

sample_from_buffer

def sample_from_buffer()

train_step

def train_step()

Module Statistics

• Classes: 4
• Functions: 12
• Imports: 8