MolecularDiffusion.modules.models.syncogen.diffusion.noise.loglinear

Log-linear noise schedule.

Classes

LogLinearNoise

Log-linear noise schedule.

Module Contents

class MolecularDiffusion.modules.models.syncogen.diffusion.noise.loglinear.LogLinearNoise(eps: float = 0.001)

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

Log-linear noise schedule.

Built such that 1 - 1/e^(n(t)) interpolates between 0 and ~1 when t varies from 0 to 1.

importance_sampling_transformation(t)

Transform timesteps for importance sampling.

rate_noise(t)

Rate of change of noise.

total_noise(t)

Total noise.

eps = 0.001
sigma_max
sigma_min