Method __call__

diff2flow/dataset/image_preprocessing.py:38–42

(self, sample)

        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"
        noise = torch.randn(self.shape)
        sample[self.key] = noise
        return sample