>>> noisy_sample = noisy_sample.to("cuda") Now create a denoising loop that predicts the residual of the less noisy sample, and computes the less noisy sample with the scheduler:   Copied >>> import tqdm