ORNL / HydraGNN

Distributed PyTorch implementation of multi-headed graph convolutional neural networks
BSD 3-Clause "New" or "Revised" License
68 stars 29 forks source link

Forces test #283

Closed RylieWeaver closed 2 months ago

RylieWeaver commented 2 months ago

This conducts force prediction tests. It directly pulls from the Lennard Jones example, rather than creating its own data independently as done in test_graphs.py. The model stacks that work with this are SchNet, EGNN, DimeNet, and PNAPlus.