Utils

TimeDirection

Bases: Enum

Enum for the direction of the noise schedule.

Source code in bionemo/moco/schedules/utils.py

class TimeDirection(Enum):
    """Enum for the direction of the noise schedule."""

    UNIFIED = "unified"  # Noise(0) --> Data(1)
    DIFFUSION = "diffusion"  # Noise(1) --> Data(0)