Torchebm > Core > Base_trainer

Contents

Classes

• BaseTrainer - Base class for training energy-based models.
• ContrastiveDivergenceTrainer - Specialized trainer for contrastive divergence training of EBMs.

API Reference

torchebm.core.base_trainer