MolecularDiffusion.modules.models.tabasco.sample.noise_schedule

Classes

BaseNoiseSchedule	Interface for time-dependent noise scaling.
RatioSampleNoiseSchedule	Ratio schedule: scale = (1 - t) / (t + eps).
SampleNoiseSchedule	Inverse schedule: scale = 1 / (t + eps).
SquareSampleNoiseSchedule	Inverse-square schedule: scale = 1 / (t**2 + eps).
ZeroSampleNoiseSchedule	Schedule that always returns zero (disables additional noise).

Module Contents

class MolecularDiffusion.modules.models.tabasco.sample.noise_schedule.BaseNoiseSchedule(cutoff: float = 0.9)

Bases: abc.ABC

Interface for time-dependent noise scaling.

Args: cutoff: Timesteps above this value return zero noise.

cutoff = 0.9

class MolecularDiffusion.modules.models.tabasco.sample.noise_schedule.RatioSampleNoiseSchedule(cutoff: float = 0.9)

Bases: BaseNoiseSchedule

Ratio schedule: scale = (1 - t) / (t + eps).

Args: cutoff: Timesteps above this value return zero noise.

class MolecularDiffusion.modules.models.tabasco.sample.noise_schedule.SampleNoiseSchedule(cutoff: float = 0.9)

Bases: BaseNoiseSchedule

Inverse schedule: scale = 1 / (t + eps).

Args: cutoff: Timesteps above this value return zero noise.

class MolecularDiffusion.modules.models.tabasco.sample.noise_schedule.SquareSampleNoiseSchedule(cutoff: float = 0.9)

Bases: BaseNoiseSchedule

Inverse-square schedule: scale = 1 / (t**2 + eps).

Args: cutoff: Timesteps above this value return zero noise.

class MolecularDiffusion.modules.models.tabasco.sample.noise_schedule.ZeroSampleNoiseSchedule(cutoff: float = 0.9)

Bases: BaseNoiseSchedule

Schedule that always returns zero (disables additional noise).

Args: cutoff: Timesteps above this value return zero noise.