MolecularDiffusion.modules.models.syncogen.diffusion.noise.linear¶

Linear noise schedule.

Classes¶

LinearNoise

Linear noise schedule.

Module Contents¶

class MolecularDiffusion.modules.models.syncogen.diffusion.noise.linear.LinearNoise(sigma_min: float = 0, sigma_max: float = 10, dtype=torch.float32)¶

Bases: MolecularDiffusion.modules.models.syncogen.diffusion.noise.base.NoiseBase

Linear noise schedule.

importance_sampling_transformation(t)¶: Transform timesteps for importance sampling.

rate_noise(t)¶: Rate of change of noise.

total_noise(t)¶: Total noise.

sigma_max¶

sigma_min¶