throw EGNN into the same file as main SE3 Transformer code, and add o…

…ptional SE3 feedforwards after every EGNN block
lucidrains · Jun 6, 2021 · c8f2013 · c8f2013
1 parent f576e5a
commit c8f2013
Show file tree

Hide file tree

Showing 4 changed files with 266 additions and 293 deletions.
diff --git a/se3_transformer_pytorch/egnn.py b/se3_transformer_pytorch/egnn.py