chenruduan / OAReactDiff

An object-aware diffusion model for generating chemical reactions

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The confidence model mentioned in the article could not be found

Romarin87 opened this issue · comments

Hello, I want to try the confidence model mentioned in the article to do some tests, but I did not find it, for example, ddpm-epoch=2074-val-totloss=531.18.ckpt in generate_confidence_sample.py.

Thanks for the interest! I uploaded the confidence model to https://zenodo.org/records/11118125. It is a ConfModule object, which inherent from LightningModule. So it can be used in a similar way as the diffusion model checkpoint.