ParallelTemperingCD

Methods and Attributes

Bases: BaseContrastiveDivergence

Source code in torchebm/losses/contrastive_divergence.py

class ParallelTemperingCD(BaseContrastiveDivergence):
    def __init__(self, temps=[1.0, 0.5], k=5):
        super().__init__(k)
        self.temps = temps  # List of temperatures

temps instance-attribute

temps = temps