MolecularDiffusion.modules.models.syncogen.diffusion.noise.cosine

Cosine noise schedules.

Classes

CosineNoise	Cosine noise schedule.
CosineSqrNoise	Squared cosine noise schedule.

Module Contents

class MolecularDiffusion.modules.models.syncogen.diffusion.noise.cosine.CosineNoise(eps: float = 0.001)

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

Cosine noise schedule.

rate_noise(t)

Rate of change of noise.

total_noise(t)

Total noise.

eps = 0.001

class MolecularDiffusion.modules.models.syncogen.diffusion.noise.cosine.CosineSqrNoise(eps: float = 0.001)

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

Squared cosine noise schedule.

rate_noise(t)

Rate of change of noise.

total_noise(t)

Total noise.

eps = 0.001