NSAPH-Projects / topological-equivariant-networks

E(n)-Equivariant Topological Neural Networks
MIT License
19 stars 0 forks source link

Unique nn for each neighbor type #36

Closed ekarais closed 5 months ago

ekarais commented 5 months ago

This PR addressed #14 by introducing the feature of utilizing unique neural networks for each neighbor type in the model, along with several enhancements and bug fixes:

This effort has streamlined several aspects of our computational setup and enriched the model's capability to handle complex network structures. We are monitoring ongoing runs to evaluate the impact of these changes on model performance.