Skip to content

Use torch_geometric.utils for scatter#61

Merged
vxfung merged 1 commit intomainfrom
torchmd_fix_scatter
Jan 22, 2024
Merged

Use torch_geometric.utils for scatter#61
vxfung merged 1 commit intomainfrom
torchmd_fix_scatter

Conversation

@itsgt
Copy link
Copy Markdown
Collaborator

@itsgt itsgt commented Jan 22, 2024

Necessary for vmap, and reduces dependence on torch_scatter (though it is used for other models)

@itsgt itsgt requested a review from vxfung January 22, 2024 17:10
@vxfung vxfung merged commit 26fc3af into main Jan 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants