Discrete Scheduler

[Abc11c]

Hi, I was looking into the paper, D3PM was wondering which scheduler to choose from particularly my problem deals with discrete action space and will be using Cross entropy to predictions and target

Currently just using DDPMScheduler, works ok but I want to see if the marginal/ uniform schedulers as seen here give me a improvement

I see there is a bunch of solutions here

Any suggestions? or solutions for me to look into.

Thanks!