choderalab / modelforge

Infrastructure to implement and train NNPs
https://modelforge.readthedocs.io/en/latest/
MIT License
9 stars 4 forks source link

Add QCArchive datastructures #1

Closed wiederm closed 11 months ago

wiederm commented 11 months ago

Description

We want to provide a dataset structure for the most common datasets that are provided through the QCArchive infrastructure. These datasets should play nicely with the pytorch dataloader objects and inherit from torch.utils.data.Dataset.

Todos

Questions

Status

chrisiacovella commented 11 months ago

I think, after a few minor changes, we should just merge this PR so we can create a separate one aimed at fixing the CI workflow.