Class UNetConfig

skills/paper2code/worked/ddpm/src/model.py:36–49

U-Net configuration. Values from Appendix B (CIFAR-10 config) and official code unless marked.

@dataclass
class UNetConfig:
    """U-Net configuration.

    Values from Appendix B (CIFAR-10 config) and official code unless marked.
    """
    image_channels: int = 3           # §4 — RGB images
    base_channels: int = 128          # Appendix B — "128 base channels"
    channel_mults: tuple = (1, 2, 2, 2)  # [FROM_OFFICIAL_CODE] — channel multipliers per level
    num_res_blocks: int = 2           # [FROM_OFFICIAL_CODE] — residual blocks per resolution level
    attention_resolutions: tuple = (16,)  # Appendix B — "attention at 16×16 resolution"
    dropout: float = 0.0             # Appendix B — "dropout 0.0" for CIFAR-10
    time_embed_dim: int = 512        # [FROM_OFFICIAL_CODE] — 4 * base_channels
    num_groups: int = 32             # [FROM_OFFICIAL_CODE] — groups for GroupNorm
    image_size: int = 32             # CIFAR-10 is 32×32


# ---------------------------------------------------------------------------