question about Non-uniform time step sampling during training

[westfish]

The paper mentions the use of Non-uniform time step sampling during training, but no relevant parameters are found in the training code. How can this strategy be applied?

[resindraburiza]

Hello,

I am not the author but I think I can give you the answer.

The related line of code about non-uniform sampling can be found here

The related setting in .yaml file is 'noise_schedule' parameter.

Best.