MolecularDiffusion.modules.models.syncogen.diffusion.noise.base¶
Base noise schedule class.
Classes¶
Base noise schedule class. |
Module Contents¶
- class MolecularDiffusion.modules.models.syncogen.diffusion.noise.base.NoiseBase¶
Bases:
abc.ABC,torch.nn.ModuleBase noise schedule class.
- forward(t)¶
Get total and rate of noise at timestep t.
- abstractmethod rate_noise(t)¶
Rate of change of noise (g(t)).
- abstractmethod total_noise(t)¶
Total noise (integral of g(t) from 0 to t + g(0)).