Discrete sceduler

[Abc11c]

Model/Pipeline/Scheduler description

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

jax implementation of the same here

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!

Open source status

• [ ] The model implementation is available.
• [ ] The model weights are available (Only relevant if addition is not a scheduler).

Provide useful links for the implementation

https://github.com/google-research/google-research/blob/master/d3pm/images/diffusion_categorical.py

[DN6]

@Abc11c It would be better to ask this question in the Discussions section.

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

[yiyixuxu]

closing this now:) feel free to ask on the discussion section