Commit 07b41d7
QET FP dataset with READMD.md added(#743)
* Adding README.md for PyG TensorNet
* added missing TrajectoryObserver
* fix ruff
* improve unit tests
* convert float into Tensor
* Corrected expected values
* Corrected expected values again
* update torch<=2.9.1
* Convert PyG TensorNet Embedding and Interaction blocks into pure Torch
* Pure Pytorch TensorNet for MLIPs is added
* Fix united tests
* Refactor the PyG TensorNet components to ensure compatibility with pure PyTorch.
* Cleanup MGLDataset
* Improve the handling of stress unit in PESCalculator
* cleanup test_ase_pyg.py
* update PESCalculator unit tests
* Improve logging in PESCalculator
* fix linting tests
* fixed linting and united tests
* include_ref_charge keyword is added in MGLDataset
* Update Relaxations and Simulations using the QET Universal Potential.ipynb
Signed-off-by: Tsz Wai Ko <47970742+kenko911@users.noreply.github.com>
* fix ruff
* avoid crashing in MatCalc united tests by adding calc_charge and compute_charge in PyG Potential and PESCalculator
* QET training module added
* fix united tests for QET training
* make sure predicted and target charges into the same dimension during the training
* added backed the self.charge_weight
* change torch.vstack into torch.hstack for reference charges
* fixed the unit test
* fix the predict_structure function in QET model
* example notebook for QET potential training
* added QET FP dataset with README
* pre-commit auto-fixes
* Clarify units for QET FP dataset in README
Added units of measurement for length, energy, force, stress, and charge.
Signed-off-by: Tsz Wai Ko <47970742+kenko911@users.noreply.github.com>
---------
Signed-off-by: Tsz Wai Ko <47970742+kenko911@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>1 parent 966672b commit 07b41d7
File tree
2 files changed
+38
-0
lines changed- pretrained_models/QET-MatQ-PES
2 files changed
+38
-0
lines changedBinary file not shown.
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
0 commit comments