Noise schedule

[bkj]

In sample.py:generate_diverse_samples I see that noise_schedule="linear", but in main.py:train_image_diffusion the noise_schedule="cosine". Is it intentional to sample w/ a different noise schedule than training? Or is that a bug / typo?

Thanks!