diffusion_models.models.mnist_enc.MNISTEncoder

class diffusion_models.models.mnist_enc.MNISTEncoder(*args: Any, **kwargs: Any)[source]

Bases: Module

__init__(out_classes=10, kernel_size=3)[source]

Methods

__init__([out_classes, kernel_size])

forward(x)