Skip to content

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