Skip to content

PersistentContrastiveDivergence

Methods and Attributes

Bases: ContrastiveDivergenceBase

Source code in torchebm/losses/contrastive_divergence.py 
52
53
54
55
56
class PersistentContrastiveDivergence(ContrastiveDivergenceBase):
    def __init__(self, buffer_size=100):
        super().__init__(k=1)
        self.buffer = None  # Persistent chain state
        self.buffer_size = buffer_size

buffer instance-attribute 

buffer = None

buffer_size instance-attribute 

buffer_size = buffer_size