Use scatter_add instead of torch.scatter_add

choderalab / modelforge

Infrastructure to implement and train NNPs

https://modelforge.readthedocs.io/en/latest/

MIT License

9 stars 4 forks source link

Closed wiederm closed 1 month ago

wiederm commented 1 month ago

The torch_scatter utility allows for more readable code; I have updated the SchNet convolution calculation using the function.