linear schedule
(timesteps, max_beta=hparams.get('max_beta', 0.01))
| 165 | |
| 166 | |
| 167 | def linear_beta_schedule(timesteps, max_beta=hparams.get('max_beta', 0.01)): |
| 168 | """ |
| 169 | linear schedule |
| 170 | """ |
| 171 | betas = np.linspace(1e-4, max_beta, timesteps) |
| 172 | return betas |
| 173 | |
| 174 | |
| 175 | def cosine_beta_schedule(timesteps, s=0.008): |
nothing calls this directly
no outgoing calls
no test coverage detected