ehoogeboom / e3_diffusion_for_molecules

MIT License
408 stars 110 forks source link

Does the code use Unet? #31

Closed Littletoyone closed 7 months ago

Littletoyone commented 7 months ago

First of all thank you so much for sharing such wonderful work. But I apologize for asking this question on a personal level, is there any framework like Unet used in this work?

ehoogeboom commented 7 months ago

Thank you. Nope, I would not know how to design an effective UNet for molecule data, this uses the EGNN.