Added tests and trainer for training generic dense neural networks using libdescriptor and pytorch (as opposed to lightning).
Batched descriptor evaluation
DNN model using TorchML driver
TODO (if any)
Need to add the capability to write the trained model as a valid DUNN driver model.
Checklist
Before a pull request can be merged, the following items must be checked:
[x] Make sure your code is properly formatted. isort and black are used for this purpose. The simplest way is to use pre-commit. See instructions here.
Summary
Added tests and trainer for training generic dense neural networks using libdescriptor and pytorch (as opposed to lightning).
TODO (if any)
Checklist
Before a pull request can be merged, the following items must be checked:
Note that the CI system will run all the above checks. But it will be much more efficient if you already fix most errors prior to submitting the PR.