Method __init__

diff2flow/dataset/image_preprocessing.py:33–36

(self, key, resolution: int = 512, latent_dim: int = 4)

class AddNoiseLatent:
    def __init__(self, key, resolution: int = 512, latent_dim: int = 4):
        self.key = key
        assert resolution % 8 == 0, "Resolution must be divisible by 8"
        self.shape = (latent_dim, resolution // 8, resolution // 8)

    def __call__(self, sample):
        assert self.key not in sample, f"Key {self.key} already exists in sample"