Method __init__

model/DocDiff.py:125–127

(self, in_channels: int, out_channels: int, time_channels: int, is_noise: bool = True)

    """

    def __init__(self, in_channels: int, out_channels: int, time_channels: int, is_noise: bool = True):
        super().__init__()
        self.res = ResidualBlock(in_channels, out_channels, time_channels, is_noise=is_noise)

    def forward(self, x: torch.Tensor, t: torch.Tensor):
        x = self.res(x, t)