Torchebm > Core > Base_loss

Contents

Classes

• BaseContrastiveDivergence - Abstract base class for Contrastive Divergence (CD) based loss functions.
• BaseLoss - Abstract base class for loss functions used in energy-based models.
• BaseScoreMatching - Abstract base class for Score Matching based loss functions.

API Reference

torchebm.core.base_loss

Base Loss Classes for Energy-Based Models

This module provides abstract base classes for defining loss functions for training energy-based models (EBMs). It includes the general BaseLoss class for arbitrary loss functions and the more specialized BaseContrastiveDivergence for contrastive divergence based training methods.

Loss functions in TorchEBM are designed to work with energy functions and samplers to define the training objective for energy-based models.