search

jasonkyuyim / se3_diffusion

Implementation for SE(3) diffusion model with application to protein backbone generation
https://arxiv.org/abs/2302.02277
MIT License
305 stars 50 forks source link

Unused PyTorch modules #14

Closed amorehead closed 1 year ago

amorehead commented 1 year ago

For clarification, is it true that the following PyTorch nn.Modules are indeed unused by your model?

https://github.com/jasonkyuyim/se3_diffusion/blob/fb350b69ca4349d7b29b19044ecc8627815dafce/model/ipa_pytorch.py#L300

https://github.com/jasonkyuyim/se3_diffusion/blob/fb350b69ca4349d7b29b19044ecc8627815dafce/model/ipa_pytorch.py#L483

jasonkyuyim commented 1 year ago

Thanks. Yes these module are unused. They can be removed if memory is an issue. I added a TODO to remove this after training a checkpoint without this unused module. https://github.com/jasonkyuyim/se3_diffusion/commit/a739080859f0abd1ea216c6907b3028a7ed3c4b3