Realign the curated model repository with the v4 workspace snapshot
Browse filesConstraint: The curated model repo must stay on main while the source snapshot lives under results/al_v4
Rejected: Keep the stale MHC-inclusive publication | it no longer matched the v4 selection
Confidence: medium
Scope-risk: moderate
Directive: Rebuild the NequIP artifact binaries from v4 checkpoints if those checkpoints are later restored
Tested: git diff --check; JSON parse sanity; refreshed MACE LAMMPS exports from v4 raw files
Not-tested: Fresh NequIP binary regeneration from v4 checkpoints, which are not present in the local results tree
- README.md +15 -56
- mace-mhc/compiled/mace_mhc_4h_stagetwo-head0.model-lammps.pt +0 -3
- mace-mhc/compiled/mace_mhc_4h_stagetwo-head1.model-lammps.pt +0 -3
- mace-mhc/compiled/mace_mhc_4h_stagetwo-head2.model-lammps.pt +0 -3
- mace-mhc/compiled/mace_mhc_4h_stagetwo-head3.model-lammps.pt +0 -3
- mace-mhc/metadata.json +0 -59
- mace-mhc/raw/mace_mhc_4h_stagetwo.model +0 -3
- mace-qbc/compiled/mace_qbc_0_stagetwo.model-lammps.pt +1 -1
- mace-qbc/compiled/mace_qbc_1_stagetwo.model-lammps.pt +1 -1
- mace-qbc/compiled/mace_qbc_2_stagetwo.model-lammps.pt +1 -1
- mace-qbc/compiled/mace_qbc_3_stagetwo.model-lammps.pt +1 -1
- mace-qbc/metadata.json +28 -11
- mace-qbc/raw/mace_qbc_0_stagetwo.model +1 -1
- mace-qbc/raw/mace_qbc_1_stagetwo.model +1 -1
- mace-qbc/raw/mace_qbc_2_stagetwo.model +1 -1
- mace-qbc/raw/mace_qbc_3_stagetwo.model +1 -1
- mace-random/compiled/mace_single_0_stagetwo.model-lammps.pt +1 -1
- mace-random/metadata.json +22 -8
- mace-random/raw/mace_single_0_stagetwo.model +1 -1
- model_index.json +95 -110
- nequip-qbc/metadata.json +29 -13
- nequip-random/metadata.json +23 -9
README.md
CHANGED
|
@@ -1,63 +1,22 @@
|
|
| 1 |
-
-
|
| 2 |
-
library_name: pytorch
|
| 3 |
-
tags:
|
| 4 |
-
- mace
|
| 5 |
-
- nequip
|
| 6 |
-
- active-learning
|
| 7 |
-
- interatomic-potentials
|
| 8 |
-
- atomistic-machine-learning
|
| 9 |
-
- computational-chemistry
|
| 10 |
-
- ase
|
| 11 |
-
- lammps
|
| 12 |
-
---
|
| 13 |
-
# FYP raw and compiled active-learning checkpoints for rMD17 ethanol
|
| 14 |
|
| 15 |
-
This
|
| 16 |
-
|
| 17 |
-
|
|
|
|
| 18 |
|
| 19 |
-
|
| 20 |
-
reusing previously uploaded assets.
|
| 21 |
|
| 22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
-
|
| 25 |
-
- **Compiled MACE checkpoints**: official LAMMPS exports produced with
|
| 26 |
-
`python -m mace.cli.create_lammps_model --format libtorch`.
|
| 27 |
-
- **Raw NequIP checkpoints**: original Lightning `.ckpt` files copied from the selected iteration.
|
| 28 |
-
- **Packaged NequIP checkpoints**: official `.nequip.zip` packages produced with `nequip-package build`.
|
| 29 |
-
- **Compiled NequIP checkpoints**: official AOTInductor artifacts produced with
|
| 30 |
-
`nequip-compile --mode aotinductor --device cuda --target ase`.
|
| 31 |
-
|
| 32 |
-
## Selected bundle summary
|
| 33 |
-
|
| 34 |
-
| Bundle | Architecture | Strategy | Best seed | Final iteration | Forces MAE (meV/Å) | Energy MAE (meV) | Raw files | Packaged files | Compiled files |
|
| 35 |
-
| --- | --- | --- | ---: | ---: | ---: | ---: | ---: | ---: | ---: |
|
| 36 |
-
| mace-random | MACE | RANDOM | 2 | 9 | 9.20 | 1.71 | 1 | 0 | 1 |
|
| 37 |
-
| mace-qbc | MACE | QBC | 2 | 9 | 8.51 | 1.60 | 4 | 0 | 4 |
|
| 38 |
-
| mace-mhc | MACE | MHC | 2 | 9 | 9.97 | 2.07 | 1 | 0 | 4 |
|
| 39 |
-
| nequip-random | NequIP | RANDOM | 3 | 9 | 8.89 | 4.74 | 1 | 1 | 1 |
|
| 40 |
-
| nequip-qbc | NequIP | QBC | 1 | 9 | 7.57 | 1.48 | 3 | 3 | 3 |
|
| 41 |
-
|
| 42 |
-
## Important details
|
| 43 |
|
| 44 |
- `mace-qbc/` contains the full four-member committee for the selected QBC seed.
|
| 45 |
-
- `mace-mhc/compiled/` contains one compiled LAMMPS export per available model head.
|
| 46 |
- `nequip-qbc/` contains one raw/package/compiled triplet per selected committee member.
|
| 47 |
-
-
|
| 48 |
-
|
| 49 |
-
running the official NequIP CLI.
|
| 50 |
-
- The compiled NequIP `.nequip.pt2` artifacts are **device-specific** AOTInductor exports
|
| 51 |
-
built for `cuda` on this machine; the `.nequip.zip` packages are the more portable
|
| 52 |
-
redistribution artifact.
|
| 53 |
-
|
| 54 |
-
## Provenance
|
| 55 |
-
|
| 56 |
-
- Hugging Face repo: `https://huggingface.co/gcnwm/fyp-active-learning-models`
|
| 57 |
-
- Source thesis workspace commit: `8fea4f53dcdf581ca23a820ab1cc8de9708f7aac`
|
| 58 |
-
- Companion production-code repository: `https://github.com/gcnwm/fyp-production`
|
| 59 |
-
- Companion production-code commit: `433f9ee3872a84cfeabf19a5d64fca9277d5cf09`
|
| 60 |
-
## Citation
|
| 61 |
-
|
| 62 |
-
If you use these checkpoints, please cite the accompanying thesis project and the
|
| 63 |
-
upstream **MACE**, **NequIP**, and **rMD17** papers.
|
|
|
|
| 1 |
+
# Curated FYP Active-Learning Checkpoints
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
+
This Hugging Face model repository stores a curated export of the strongest
|
| 4 |
+
final-iteration checkpoints from the MACE and NequIP active-learning study on
|
| 5 |
+
rMD17 ethanol. Each bundle is selected by the best final forces MAE across the
|
| 6 |
+
three repeated experiment seeds for its architecture/strategy family.
|
| 7 |
|
| 8 |
+
## Included bundles
|
|
|
|
| 9 |
|
| 10 |
+
| Bundle | Architecture | Strategy | Selected seed | Final forces MAE (meV/Å) | Final energy MAE (meV) | Files |
|
| 11 |
+
| --- | --- | --- | ---: | ---: | ---: | ---: |
|
| 12 |
+
| mace-random | MACE | RANDOM | 1 | 9.45 | 1.79 | 2 |
|
| 13 |
+
| mace-qbc | MACE | QBC | 3 | 8.61 | 1.69 | 8 |
|
| 14 |
+
| nequip-random | NequIP | RANDOM | 2 | 10.66 | 4.24 | 7 |
|
| 15 |
+
| nequip-qbc | NequIP | QBC | 3 | 9.53 | 6.52 | 21 |
|
| 16 |
|
| 17 |
+
## Notes
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
- `mace-qbc/` contains the full four-member committee for the selected QBC seed.
|
|
|
|
| 20 |
- `nequip-qbc/` contains one raw/package/compiled triplet per selected committee member.
|
| 21 |
+
- Per-bundle `metadata.json` files record the selected seed, final metrics, and
|
| 22 |
+
original relative source paths from the thesis workspace.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
mace-mhc/compiled/mace_mhc_4h_stagetwo-head0.model-lammps.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:ef2ef5bc7d39e662341c06d40f5cf2562bd4ebc3fab9a0659de7baa6cfed2b1e
|
| 3 |
-
size 7737953
|
|
|
|
|
|
|
|
|
|
|
|
mace-mhc/compiled/mace_mhc_4h_stagetwo-head1.model-lammps.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:b80edbb41eba2f9173642837ca7af1571dc7d77a4433ddd98288f8bdf562d21c
|
| 3 |
-
size 7737953
|
|
|
|
|
|
|
|
|
|
|
|
mace-mhc/compiled/mace_mhc_4h_stagetwo-head2.model-lammps.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:4884c7cd61f90551206019a59cba9dd832fae0185a6caae35af6dd9444dab5ff
|
| 3 |
-
size 7737953
|
|
|
|
|
|
|
|
|
|
|
|
mace-mhc/compiled/mace_mhc_4h_stagetwo-head3.model-lammps.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:d3978bf52d35e94d1b6cae79ea27a9bbe5aea4f82fb6228ef45d79e18da9ad02
|
| 3 |
-
size 7737953
|
|
|
|
|
|
|
|
|
|
|
|
mace-mhc/metadata.json
DELETED
|
@@ -1,59 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"architecture": "MACE",
|
| 3 |
-
"bundle": "mace-mhc",
|
| 4 |
-
"compiled_files": [
|
| 5 |
-
{
|
| 6 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head0 --format libtorch",
|
| 7 |
-
"format": "libtorch",
|
| 8 |
-
"head": "head0",
|
| 9 |
-
"published_file": "mace_mhc_4h_stagetwo-head0.model-lammps.pt",
|
| 10 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 11 |
-
},
|
| 12 |
-
{
|
| 13 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head1 --format libtorch",
|
| 14 |
-
"format": "libtorch",
|
| 15 |
-
"head": "head1",
|
| 16 |
-
"published_file": "mace_mhc_4h_stagetwo-head1.model-lammps.pt",
|
| 17 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 18 |
-
},
|
| 19 |
-
{
|
| 20 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head2 --format libtorch",
|
| 21 |
-
"format": "libtorch",
|
| 22 |
-
"head": "head2",
|
| 23 |
-
"published_file": "mace_mhc_4h_stagetwo-head2.model-lammps.pt",
|
| 24 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 25 |
-
},
|
| 26 |
-
{
|
| 27 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head3 --format libtorch",
|
| 28 |
-
"format": "libtorch",
|
| 29 |
-
"head": "head3",
|
| 30 |
-
"published_file": "mace_mhc_4h_stagetwo-head3.model-lammps.pt",
|
| 31 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 32 |
-
}
|
| 33 |
-
],
|
| 34 |
-
"final_iteration": 9,
|
| 35 |
-
"final_metrics": {
|
| 36 |
-
"energy_mae": 0.00207461045447144,
|
| 37 |
-
"energy_rmse": 0.002818075833732281,
|
| 38 |
-
"forces_mae": 0.00997466210127421,
|
| 39 |
-
"forces_rmse": 0.014573431573543202,
|
| 40 |
-
"iteration": 9,
|
| 41 |
-
"max_disagreement": 0.0,
|
| 42 |
-
"mean_disagreement": 0.0,
|
| 43 |
-
"n_labeled": 550,
|
| 44 |
-
"training_time": 2519.816365480423
|
| 45 |
-
},
|
| 46 |
-
"packaged_files": [],
|
| 47 |
-
"raw_files": [
|
| 48 |
-
{
|
| 49 |
-
"published_file": "mace_mhc_4h_stagetwo.model",
|
| 50 |
-
"source_file": "results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
|
| 51 |
-
}
|
| 52 |
-
],
|
| 53 |
-
"selected_seed": 2,
|
| 54 |
-
"source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
|
| 55 |
-
"source_files": [
|
| 56 |
-
"results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
|
| 57 |
-
],
|
| 58 |
-
"strategy": "mhc"
|
| 59 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
mace-mhc/raw/mace_mhc_4h_stagetwo.model
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:1ba54febe19dc369f833513bbfa5cee4d7ca6374fc79caa66b05c958a4fd41fa
|
| 3 |
-
size 7639289
|
|
|
|
|
|
|
|
|
|
|
|
mace-qbc/compiled/mace_qbc_0_stagetwo.model-lammps.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7722036
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:42d642e82691b6b50d2bf26842a6427ea8b3f1c608af072472f92cacff1497c1
|
| 3 |
size 7722036
|
mace-qbc/compiled/mace_qbc_1_stagetwo.model-lammps.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7722036
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:01cfb69fec7746f5c2fe56811d02592fdfc816d4706bced95a0ab9e9817ed3fa
|
| 3 |
size 7722036
|
mace-qbc/compiled/mace_qbc_2_stagetwo.model-lammps.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7722036
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b25b1c50a87a3e3e07dd4abe04477f9e9a07205fe906d5e430ecfec37309e6f6
|
| 3 |
size 7722036
|
mace-qbc/compiled/mace_qbc_3_stagetwo.model-lammps.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7722036
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d48f7209a223d052289e0255a05520b2d1de058136ab45f3fa2206456cb0b127
|
| 3 |
size 7722036
|
mace-qbc/metadata.json
CHANGED
|
@@ -33,17 +33,34 @@
|
|
| 33 |
],
|
| 34 |
"final_iteration": 9,
|
| 35 |
"final_metrics": {
|
| 36 |
-
"
|
| 37 |
-
"
|
| 38 |
-
"
|
| 39 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
"iteration": 9,
|
|
|
|
| 41 |
"max_disagreement": 0.0,
|
| 42 |
"mean_disagreement": 0.0,
|
| 43 |
"n_labeled": 550,
|
| 44 |
-
"
|
|
|
|
|
|
|
|
|
|
| 45 |
},
|
| 46 |
"packaged_files": [],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
"raw_files": [
|
| 48 |
{
|
| 49 |
"published_file": "mace_qbc_0_stagetwo.model",
|
|
@@ -62,13 +79,13 @@
|
|
| 62 |
"source_file": "results/al/mace_qbc_seed2/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
|
| 63 |
}
|
| 64 |
],
|
| 65 |
-
"selected_seed":
|
| 66 |
-
"source_commit": "
|
| 67 |
"source_files": [
|
| 68 |
-
"results/
|
| 69 |
-
"results/
|
| 70 |
-
"results/
|
| 71 |
-
"results/
|
| 72 |
],
|
| 73 |
"strategy": "qbc"
|
| 74 |
}
|
|
|
|
| 33 |
],
|
| 34 |
"final_iteration": 9,
|
| 35 |
"final_metrics": {
|
| 36 |
+
"acquisition_rule": "uncertainty",
|
| 37 |
+
"acquisition_seed": null,
|
| 38 |
+
"candidate_n": 0,
|
| 39 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 40 |
+
"candidate_seed": null,
|
| 41 |
+
"energy_mae": 0.001692634182771144,
|
| 42 |
+
"energy_rmse": 0.0022133917348856097,
|
| 43 |
+
"evaluation_policy": "fixed_member",
|
| 44 |
+
"forces_mae": 0.008605168772635183,
|
| 45 |
+
"forces_rmse": 0.01183799714274852,
|
| 46 |
+
"is_acquisition_iteration": false,
|
| 47 |
"iteration": 9,
|
| 48 |
+
"labeled_n_after_acquisition": 550,
|
| 49 |
"max_disagreement": 0.0,
|
| 50 |
"mean_disagreement": 0.0,
|
| 51 |
"n_labeled": 550,
|
| 52 |
+
"selected_n": 0,
|
| 53 |
+
"train_n_before_acquisition": 550,
|
| 54 |
+
"training_policy": "scratch",
|
| 55 |
+
"training_time": 6118.126273155212
|
| 56 |
},
|
| 57 |
"packaged_files": [],
|
| 58 |
+
"published_files": [
|
| 59 |
+
"mace_qbc_0_stagetwo.model",
|
| 60 |
+
"mace_qbc_1_stagetwo.model",
|
| 61 |
+
"mace_qbc_2_stagetwo.model",
|
| 62 |
+
"mace_qbc_3_stagetwo.model"
|
| 63 |
+
],
|
| 64 |
"raw_files": [
|
| 65 |
{
|
| 66 |
"published_file": "mace_qbc_0_stagetwo.model",
|
|
|
|
| 79 |
"source_file": "results/al/mace_qbc_seed2/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
|
| 80 |
}
|
| 81 |
],
|
| 82 |
+
"selected_seed": 3,
|
| 83 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 84 |
"source_files": [
|
| 85 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model",
|
| 86 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model",
|
| 87 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model",
|
| 88 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
|
| 89 |
],
|
| 90 |
"strategy": "qbc"
|
| 91 |
}
|
mace-qbc/raw/mace_qbc_0_stagetwo.model
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7623775
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:96b85694e2869521530dcb9b22835693ce34c3a81af2fc5911167205c1d5d70d
|
| 3 |
size 7623775
|
mace-qbc/raw/mace_qbc_1_stagetwo.model
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7623775
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d39617aa283c96832ed0a8c8a55e3bc51b82e6a1059869278ba89fd6ac7014d1
|
| 3 |
size 7623775
|
mace-qbc/raw/mace_qbc_2_stagetwo.model
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7623775
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4bde3567e7e7b1a4e56a34954ca7bb2d76bbabfbcf104213c14225a63448f53a
|
| 3 |
size 7623775
|
mace-qbc/raw/mace_qbc_3_stagetwo.model
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7623775
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b206ba224196a6f90e59a1adaae677bb3cab9f1fc4e8355c0af3e8b081cff5a
|
| 3 |
size 7623775
|
mace-random/compiled/mace_single_0_stagetwo.model-lammps.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7723131
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0f1b479bbb6bcb7c7b5589796187729f73488e2540386865d258d0d35dc6d9dd
|
| 3 |
size 7723131
|
mace-random/metadata.json
CHANGED
|
@@ -12,27 +12,41 @@
|
|
| 12 |
],
|
| 13 |
"final_iteration": 9,
|
| 14 |
"final_metrics": {
|
| 15 |
-
"
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
"iteration": 9,
|
|
|
|
| 20 |
"max_disagreement": 0.0,
|
| 21 |
"mean_disagreement": 0.0,
|
| 22 |
"n_labeled": 550,
|
| 23 |
-
"
|
|
|
|
|
|
|
|
|
|
| 24 |
},
|
| 25 |
"packaged_files": [],
|
|
|
|
|
|
|
|
|
|
| 26 |
"raw_files": [
|
| 27 |
{
|
| 28 |
"published_file": "mace_single_0_stagetwo.model",
|
| 29 |
"source_file": "results/al/random_seed2/iter09_model/models/mace_single_0_stagetwo.model"
|
| 30 |
}
|
| 31 |
],
|
| 32 |
-
"selected_seed":
|
| 33 |
-
"source_commit": "
|
| 34 |
"source_files": [
|
| 35 |
-
"results/
|
| 36 |
],
|
| 37 |
"strategy": "random"
|
| 38 |
}
|
|
|
|
| 12 |
],
|
| 13 |
"final_iteration": 9,
|
| 14 |
"final_metrics": {
|
| 15 |
+
"acquisition_rule": "random",
|
| 16 |
+
"acquisition_seed": null,
|
| 17 |
+
"candidate_n": 0,
|
| 18 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 19 |
+
"candidate_seed": null,
|
| 20 |
+
"energy_mae": 0.0017917274611536414,
|
| 21 |
+
"energy_rmse": 0.002695636120741299,
|
| 22 |
+
"evaluation_policy": "fixed_member",
|
| 23 |
+
"forces_mae": 0.009449580705448683,
|
| 24 |
+
"forces_rmse": 0.014669491447455139,
|
| 25 |
+
"is_acquisition_iteration": false,
|
| 26 |
"iteration": 9,
|
| 27 |
+
"labeled_n_after_acquisition": 550,
|
| 28 |
"max_disagreement": 0.0,
|
| 29 |
"mean_disagreement": 0.0,
|
| 30 |
"n_labeled": 550,
|
| 31 |
+
"selected_n": 0,
|
| 32 |
+
"train_n_before_acquisition": 550,
|
| 33 |
+
"training_policy": "scratch",
|
| 34 |
+
"training_time": 1608.996175289154
|
| 35 |
},
|
| 36 |
"packaged_files": [],
|
| 37 |
+
"published_files": [
|
| 38 |
+
"mace_single_0_stagetwo.model"
|
| 39 |
+
],
|
| 40 |
"raw_files": [
|
| 41 |
{
|
| 42 |
"published_file": "mace_single_0_stagetwo.model",
|
| 43 |
"source_file": "results/al/random_seed2/iter09_model/models/mace_single_0_stagetwo.model"
|
| 44 |
}
|
| 45 |
],
|
| 46 |
+
"selected_seed": 1,
|
| 47 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 48 |
"source_files": [
|
| 49 |
+
"results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model"
|
| 50 |
],
|
| 51 |
"strategy": "random"
|
| 52 |
}
|
mace-random/raw/mace_single_0_stagetwo.model
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7624109
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:011f6433dff21de38318de0d8e86c25a84529a4e065d1ba038a9c65dcafb89d0
|
| 3 |
size 7624109
|
model_index.json
CHANGED
|
@@ -4,7 +4,7 @@
|
|
| 4 |
"bundle": "mace-random",
|
| 5 |
"compiled_files": [
|
| 6 |
{
|
| 7 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /
|
| 8 |
"format": "libtorch",
|
| 9 |
"head": "Default",
|
| 10 |
"published_file": "mace_single_0_stagetwo.model-lammps.pt",
|
|
@@ -13,27 +13,38 @@
|
|
| 13 |
],
|
| 14 |
"final_iteration": 9,
|
| 15 |
"final_metrics": {
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
-
"
|
| 19 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
"iteration": 9,
|
|
|
|
| 21 |
"max_disagreement": 0.0,
|
| 22 |
"mean_disagreement": 0.0,
|
| 23 |
"n_labeled": 550,
|
| 24 |
-
"
|
|
|
|
|
|
|
|
|
|
| 25 |
},
|
| 26 |
"packaged_files": [],
|
| 27 |
"raw_files": [
|
| 28 |
{
|
| 29 |
"published_file": "mace_single_0_stagetwo.model",
|
| 30 |
-
"source_file": "results/
|
| 31 |
}
|
| 32 |
],
|
| 33 |
-
"selected_seed":
|
| 34 |
-
"source_commit": "
|
| 35 |
"source_files": [
|
| 36 |
-
"results/
|
| 37 |
],
|
| 38 |
"strategy": "random"
|
| 39 |
},
|
|
@@ -42,28 +53,28 @@
|
|
| 42 |
"bundle": "mace-qbc",
|
| 43 |
"compiled_files": [
|
| 44 |
{
|
| 45 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /
|
| 46 |
"format": "libtorch",
|
| 47 |
"head": "Default",
|
| 48 |
"published_file": "mace_qbc_0_stagetwo.model-lammps.pt",
|
| 49 |
"raw_file": "mace_qbc_0_stagetwo.model"
|
| 50 |
},
|
| 51 |
{
|
| 52 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /
|
| 53 |
"format": "libtorch",
|
| 54 |
"head": "Default",
|
| 55 |
"published_file": "mace_qbc_1_stagetwo.model-lammps.pt",
|
| 56 |
"raw_file": "mace_qbc_1_stagetwo.model"
|
| 57 |
},
|
| 58 |
{
|
| 59 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /
|
| 60 |
"format": "libtorch",
|
| 61 |
"head": "Default",
|
| 62 |
"published_file": "mace_qbc_2_stagetwo.model-lammps.pt",
|
| 63 |
"raw_file": "mace_qbc_2_stagetwo.model"
|
| 64 |
},
|
| 65 |
{
|
| 66 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /
|
| 67 |
"format": "libtorch",
|
| 68 |
"head": "Default",
|
| 69 |
"published_file": "mace_qbc_3_stagetwo.model-lammps.pt",
|
|
@@ -72,104 +83,56 @@
|
|
| 72 |
],
|
| 73 |
"final_iteration": 9,
|
| 74 |
"final_metrics": {
|
| 75 |
-
"
|
| 76 |
-
"
|
| 77 |
-
"
|
| 78 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 79 |
"iteration": 9,
|
|
|
|
| 80 |
"max_disagreement": 0.0,
|
| 81 |
"mean_disagreement": 0.0,
|
| 82 |
"n_labeled": 550,
|
| 83 |
-
"
|
|
|
|
|
|
|
|
|
|
| 84 |
},
|
| 85 |
"packaged_files": [],
|
| 86 |
"raw_files": [
|
| 87 |
{
|
| 88 |
"published_file": "mace_qbc_0_stagetwo.model",
|
| 89 |
-
"source_file": "results/
|
| 90 |
},
|
| 91 |
{
|
| 92 |
"published_file": "mace_qbc_1_stagetwo.model",
|
| 93 |
-
"source_file": "results/
|
| 94 |
},
|
| 95 |
{
|
| 96 |
"published_file": "mace_qbc_2_stagetwo.model",
|
| 97 |
-
"source_file": "results/
|
| 98 |
},
|
| 99 |
{
|
| 100 |
"published_file": "mace_qbc_3_stagetwo.model",
|
| 101 |
-
"source_file": "results/
|
| 102 |
}
|
| 103 |
],
|
| 104 |
-
"selected_seed":
|
| 105 |
-
"source_commit": "
|
| 106 |
"source_files": [
|
| 107 |
-
"results/
|
| 108 |
-
"results/
|
| 109 |
-
"results/
|
| 110 |
-
"results/
|
| 111 |
],
|
| 112 |
"strategy": "qbc"
|
| 113 |
},
|
| 114 |
-
{
|
| 115 |
-
"architecture": "MACE",
|
| 116 |
-
"bundle": "mace-mhc",
|
| 117 |
-
"compiled_files": [
|
| 118 |
-
{
|
| 119 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head0 --format libtorch",
|
| 120 |
-
"format": "libtorch",
|
| 121 |
-
"head": "head0",
|
| 122 |
-
"published_file": "mace_mhc_4h_stagetwo-head0.model-lammps.pt",
|
| 123 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 124 |
-
},
|
| 125 |
-
{
|
| 126 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head1 --format libtorch",
|
| 127 |
-
"format": "libtorch",
|
| 128 |
-
"head": "head1",
|
| 129 |
-
"published_file": "mace_mhc_4h_stagetwo-head1.model-lammps.pt",
|
| 130 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 131 |
-
},
|
| 132 |
-
{
|
| 133 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head2 --format libtorch",
|
| 134 |
-
"format": "libtorch",
|
| 135 |
-
"head": "head2",
|
| 136 |
-
"published_file": "mace_mhc_4h_stagetwo-head2.model-lammps.pt",
|
| 137 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 138 |
-
},
|
| 139 |
-
{
|
| 140 |
-
"command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head3 --format libtorch",
|
| 141 |
-
"format": "libtorch",
|
| 142 |
-
"head": "head3",
|
| 143 |
-
"published_file": "mace_mhc_4h_stagetwo-head3.model-lammps.pt",
|
| 144 |
-
"raw_file": "mace_mhc_4h_stagetwo.model"
|
| 145 |
-
}
|
| 146 |
-
],
|
| 147 |
-
"final_iteration": 9,
|
| 148 |
-
"final_metrics": {
|
| 149 |
-
"energy_mae": 0.00207461045447144,
|
| 150 |
-
"energy_rmse": 0.002818075833732281,
|
| 151 |
-
"forces_mae": 0.00997466210127421,
|
| 152 |
-
"forces_rmse": 0.014573431573543202,
|
| 153 |
-
"iteration": 9,
|
| 154 |
-
"max_disagreement": 0.0,
|
| 155 |
-
"mean_disagreement": 0.0,
|
| 156 |
-
"n_labeled": 550,
|
| 157 |
-
"training_time": 2519.816365480423
|
| 158 |
-
},
|
| 159 |
-
"packaged_files": [],
|
| 160 |
-
"raw_files": [
|
| 161 |
-
{
|
| 162 |
-
"published_file": "mace_mhc_4h_stagetwo.model",
|
| 163 |
-
"source_file": "results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
|
| 164 |
-
}
|
| 165 |
-
],
|
| 166 |
-
"selected_seed": 2,
|
| 167 |
-
"source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
|
| 168 |
-
"source_files": [
|
| 169 |
-
"results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
|
| 170 |
-
],
|
| 171 |
-
"strategy": "mhc"
|
| 172 |
-
},
|
| 173 |
{
|
| 174 |
"architecture": "NequIP",
|
| 175 |
"bundle": "nequip-random",
|
|
@@ -184,15 +147,26 @@
|
|
| 184 |
],
|
| 185 |
"final_iteration": 9,
|
| 186 |
"final_metrics": {
|
| 187 |
-
"
|
| 188 |
-
"
|
| 189 |
-
"
|
| 190 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 191 |
"iteration": 9,
|
|
|
|
| 192 |
"max_disagreement": 0.0,
|
| 193 |
"mean_disagreement": 0.0,
|
| 194 |
"n_labeled": 550,
|
| 195 |
-
"
|
|
|
|
|
|
|
|
|
|
| 196 |
},
|
| 197 |
"packaged_files": [
|
| 198 |
{
|
|
@@ -204,13 +178,13 @@
|
|
| 204 |
"raw_files": [
|
| 205 |
{
|
| 206 |
"published_file": "nequip_single_0-best.ckpt",
|
| 207 |
-
"source_file": "results/
|
| 208 |
}
|
| 209 |
],
|
| 210 |
-
"selected_seed":
|
| 211 |
-
"source_commit": "
|
| 212 |
"source_files": [
|
| 213 |
-
"results/
|
| 214 |
],
|
| 215 |
"strategy": "random"
|
| 216 |
},
|
|
@@ -242,15 +216,26 @@
|
|
| 242 |
],
|
| 243 |
"final_iteration": 9,
|
| 244 |
"final_metrics": {
|
| 245 |
-
"
|
| 246 |
-
"
|
| 247 |
-
"
|
| 248 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 249 |
"iteration": 9,
|
|
|
|
| 250 |
"max_disagreement": 0.0,
|
| 251 |
"mean_disagreement": 0.0,
|
| 252 |
"n_labeled": 550,
|
| 253 |
-
"
|
|
|
|
|
|
|
|
|
|
| 254 |
},
|
| 255 |
"packaged_files": [
|
| 256 |
{
|
|
@@ -272,23 +257,23 @@
|
|
| 272 |
"raw_files": [
|
| 273 |
{
|
| 274 |
"published_file": "committee_0-best.ckpt",
|
| 275 |
-
"source_file": "results/
|
| 276 |
},
|
| 277 |
{
|
| 278 |
"published_file": "committee_1-best.ckpt",
|
| 279 |
-
"source_file": "results/
|
| 280 |
},
|
| 281 |
{
|
| 282 |
"published_file": "committee_2-best.ckpt",
|
| 283 |
-
"source_file": "results/
|
| 284 |
}
|
| 285 |
],
|
| 286 |
-
"selected_seed":
|
| 287 |
-
"source_commit": "
|
| 288 |
"source_files": [
|
| 289 |
-
"results/
|
| 290 |
-
"results/
|
| 291 |
-
"results/
|
| 292 |
],
|
| 293 |
"strategy": "qbc"
|
| 294 |
}
|
|
|
|
| 4 |
"bundle": "mace-random",
|
| 5 |
"compiled_files": [
|
| 6 |
{
|
| 7 |
+
"command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model --format libtorch",
|
| 8 |
"format": "libtorch",
|
| 9 |
"head": "Default",
|
| 10 |
"published_file": "mace_single_0_stagetwo.model-lammps.pt",
|
|
|
|
| 13 |
],
|
| 14 |
"final_iteration": 9,
|
| 15 |
"final_metrics": {
|
| 16 |
+
"acquisition_rule": "random",
|
| 17 |
+
"acquisition_seed": null,
|
| 18 |
+
"candidate_n": 0,
|
| 19 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 20 |
+
"candidate_seed": null,
|
| 21 |
+
"energy_mae": 0.0017917274611536414,
|
| 22 |
+
"energy_rmse": 0.002695636120741299,
|
| 23 |
+
"evaluation_policy": "fixed_member",
|
| 24 |
+
"forces_mae": 0.009449580705448683,
|
| 25 |
+
"forces_rmse": 0.014669491447455139,
|
| 26 |
+
"is_acquisition_iteration": false,
|
| 27 |
"iteration": 9,
|
| 28 |
+
"labeled_n_after_acquisition": 550,
|
| 29 |
"max_disagreement": 0.0,
|
| 30 |
"mean_disagreement": 0.0,
|
| 31 |
"n_labeled": 550,
|
| 32 |
+
"selected_n": 0,
|
| 33 |
+
"train_n_before_acquisition": 550,
|
| 34 |
+
"training_policy": "scratch",
|
| 35 |
+
"training_time": 1608.996175289154
|
| 36 |
},
|
| 37 |
"packaged_files": [],
|
| 38 |
"raw_files": [
|
| 39 |
{
|
| 40 |
"published_file": "mace_single_0_stagetwo.model",
|
| 41 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model"
|
| 42 |
}
|
| 43 |
],
|
| 44 |
+
"selected_seed": 1,
|
| 45 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 46 |
"source_files": [
|
| 47 |
+
"results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model"
|
| 48 |
],
|
| 49 |
"strategy": "random"
|
| 50 |
},
|
|
|
|
| 53 |
"bundle": "mace-qbc",
|
| 54 |
"compiled_files": [
|
| 55 |
{
|
| 56 |
+
"command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model --format libtorch",
|
| 57 |
"format": "libtorch",
|
| 58 |
"head": "Default",
|
| 59 |
"published_file": "mace_qbc_0_stagetwo.model-lammps.pt",
|
| 60 |
"raw_file": "mace_qbc_0_stagetwo.model"
|
| 61 |
},
|
| 62 |
{
|
| 63 |
+
"command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model --format libtorch",
|
| 64 |
"format": "libtorch",
|
| 65 |
"head": "Default",
|
| 66 |
"published_file": "mace_qbc_1_stagetwo.model-lammps.pt",
|
| 67 |
"raw_file": "mace_qbc_1_stagetwo.model"
|
| 68 |
},
|
| 69 |
{
|
| 70 |
+
"command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model --format libtorch",
|
| 71 |
"format": "libtorch",
|
| 72 |
"head": "Default",
|
| 73 |
"published_file": "mace_qbc_2_stagetwo.model-lammps.pt",
|
| 74 |
"raw_file": "mace_qbc_2_stagetwo.model"
|
| 75 |
},
|
| 76 |
{
|
| 77 |
+
"command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model --format libtorch",
|
| 78 |
"format": "libtorch",
|
| 79 |
"head": "Default",
|
| 80 |
"published_file": "mace_qbc_3_stagetwo.model-lammps.pt",
|
|
|
|
| 83 |
],
|
| 84 |
"final_iteration": 9,
|
| 85 |
"final_metrics": {
|
| 86 |
+
"acquisition_rule": "uncertainty",
|
| 87 |
+
"acquisition_seed": null,
|
| 88 |
+
"candidate_n": 0,
|
| 89 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 90 |
+
"candidate_seed": null,
|
| 91 |
+
"energy_mae": 0.001692634182771144,
|
| 92 |
+
"energy_rmse": 0.0022133917348856097,
|
| 93 |
+
"evaluation_policy": "fixed_member",
|
| 94 |
+
"forces_mae": 0.008605168772635183,
|
| 95 |
+
"forces_rmse": 0.01183799714274852,
|
| 96 |
+
"is_acquisition_iteration": false,
|
| 97 |
"iteration": 9,
|
| 98 |
+
"labeled_n_after_acquisition": 550,
|
| 99 |
"max_disagreement": 0.0,
|
| 100 |
"mean_disagreement": 0.0,
|
| 101 |
"n_labeled": 550,
|
| 102 |
+
"selected_n": 0,
|
| 103 |
+
"train_n_before_acquisition": 550,
|
| 104 |
+
"training_policy": "scratch",
|
| 105 |
+
"training_time": 6118.126273155212
|
| 106 |
},
|
| 107 |
"packaged_files": [],
|
| 108 |
"raw_files": [
|
| 109 |
{
|
| 110 |
"published_file": "mace_qbc_0_stagetwo.model",
|
| 111 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model"
|
| 112 |
},
|
| 113 |
{
|
| 114 |
"published_file": "mace_qbc_1_stagetwo.model",
|
| 115 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model"
|
| 116 |
},
|
| 117 |
{
|
| 118 |
"published_file": "mace_qbc_2_stagetwo.model",
|
| 119 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model"
|
| 120 |
},
|
| 121 |
{
|
| 122 |
"published_file": "mace_qbc_3_stagetwo.model",
|
| 123 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
|
| 124 |
}
|
| 125 |
],
|
| 126 |
+
"selected_seed": 3,
|
| 127 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 128 |
"source_files": [
|
| 129 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model",
|
| 130 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model",
|
| 131 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model",
|
| 132 |
+
"results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
|
| 133 |
],
|
| 134 |
"strategy": "qbc"
|
| 135 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 136 |
{
|
| 137 |
"architecture": "NequIP",
|
| 138 |
"bundle": "nequip-random",
|
|
|
|
| 147 |
],
|
| 148 |
"final_iteration": 9,
|
| 149 |
"final_metrics": {
|
| 150 |
+
"acquisition_rule": "random",
|
| 151 |
+
"acquisition_seed": null,
|
| 152 |
+
"candidate_n": 0,
|
| 153 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 154 |
+
"candidate_seed": null,
|
| 155 |
+
"energy_mae": 0.004235895300860648,
|
| 156 |
+
"energy_rmse": 0.0050607236903155136,
|
| 157 |
+
"evaluation_policy": "fixed_member",
|
| 158 |
+
"forces_mae": 0.01065873935388155,
|
| 159 |
+
"forces_rmse": 0.017322629771456894,
|
| 160 |
+
"is_acquisition_iteration": false,
|
| 161 |
"iteration": 9,
|
| 162 |
+
"labeled_n_after_acquisition": 550,
|
| 163 |
"max_disagreement": 0.0,
|
| 164 |
"mean_disagreement": 0.0,
|
| 165 |
"n_labeled": 550,
|
| 166 |
+
"selected_n": 0,
|
| 167 |
+
"train_n_before_acquisition": 550,
|
| 168 |
+
"training_policy": "scratch",
|
| 169 |
+
"training_time": 4275.6661376953125
|
| 170 |
},
|
| 171 |
"packaged_files": [
|
| 172 |
{
|
|
|
|
| 178 |
"raw_files": [
|
| 179 |
{
|
| 180 |
"published_file": "nequip_single_0-best.ckpt",
|
| 181 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
|
| 182 |
}
|
| 183 |
],
|
| 184 |
+
"selected_seed": 2,
|
| 185 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 186 |
"source_files": [
|
| 187 |
+
"results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
|
| 188 |
],
|
| 189 |
"strategy": "random"
|
| 190 |
},
|
|
|
|
| 216 |
],
|
| 217 |
"final_iteration": 9,
|
| 218 |
"final_metrics": {
|
| 219 |
+
"acquisition_rule": "uncertainty",
|
| 220 |
+
"acquisition_seed": null,
|
| 221 |
+
"candidate_n": 0,
|
| 222 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 223 |
+
"candidate_seed": null,
|
| 224 |
+
"energy_mae": 0.006517067205983039,
|
| 225 |
+
"energy_rmse": 0.007017354304601934,
|
| 226 |
+
"evaluation_policy": "fixed_member",
|
| 227 |
+
"forces_mae": 0.009533511264421878,
|
| 228 |
+
"forces_rmse": 0.013129118546651844,
|
| 229 |
+
"is_acquisition_iteration": false,
|
| 230 |
"iteration": 9,
|
| 231 |
+
"labeled_n_after_acquisition": 550,
|
| 232 |
"max_disagreement": 0.0,
|
| 233 |
"mean_disagreement": 0.0,
|
| 234 |
"n_labeled": 550,
|
| 235 |
+
"selected_n": 0,
|
| 236 |
+
"train_n_before_acquisition": 550,
|
| 237 |
+
"training_policy": "scratch",
|
| 238 |
+
"training_time": 17430.808257341385
|
| 239 |
},
|
| 240 |
"packaged_files": [
|
| 241 |
{
|
|
|
|
| 257 |
"raw_files": [
|
| 258 |
{
|
| 259 |
"published_file": "committee_0-best.ckpt",
|
| 260 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt"
|
| 261 |
},
|
| 262 |
{
|
| 263 |
"published_file": "committee_1-best.ckpt",
|
| 264 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt"
|
| 265 |
},
|
| 266 |
{
|
| 267 |
"published_file": "committee_2-best.ckpt",
|
| 268 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
|
| 269 |
}
|
| 270 |
],
|
| 271 |
+
"selected_seed": 3,
|
| 272 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 273 |
"source_files": [
|
| 274 |
+
"results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt",
|
| 275 |
+
"results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt",
|
| 276 |
+
"results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
|
| 277 |
],
|
| 278 |
"strategy": "qbc"
|
| 279 |
}
|
nequip-qbc/metadata.json
CHANGED
|
@@ -26,15 +26,26 @@
|
|
| 26 |
],
|
| 27 |
"final_iteration": 9,
|
| 28 |
"final_metrics": {
|
| 29 |
-
"
|
| 30 |
-
"
|
| 31 |
-
"
|
| 32 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
"iteration": 9,
|
|
|
|
| 34 |
"max_disagreement": 0.0,
|
| 35 |
"mean_disagreement": 0.0,
|
| 36 |
"n_labeled": 550,
|
| 37 |
-
"
|
|
|
|
|
|
|
|
|
|
| 38 |
},
|
| 39 |
"packaged_files": [
|
| 40 |
{
|
|
@@ -53,26 +64,31 @@
|
|
| 53 |
"raw_file": "committee_2-best.ckpt"
|
| 54 |
}
|
| 55 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
"raw_files": [
|
| 57 |
{
|
| 58 |
"published_file": "committee_0-best.ckpt",
|
| 59 |
-
"source_file": "results/
|
| 60 |
},
|
| 61 |
{
|
| 62 |
"published_file": "committee_1-best.ckpt",
|
| 63 |
-
"source_file": "results/
|
| 64 |
},
|
| 65 |
{
|
| 66 |
"published_file": "committee_2-best.ckpt",
|
| 67 |
-
"source_file": "results/
|
| 68 |
}
|
| 69 |
],
|
| 70 |
-
"selected_seed":
|
| 71 |
-
"source_commit": "
|
| 72 |
"source_files": [
|
| 73 |
-
"results/
|
| 74 |
-
"results/
|
| 75 |
-
"results/
|
| 76 |
],
|
| 77 |
"strategy": "qbc"
|
| 78 |
}
|
|
|
|
| 26 |
],
|
| 27 |
"final_iteration": 9,
|
| 28 |
"final_metrics": {
|
| 29 |
+
"acquisition_rule": "uncertainty",
|
| 30 |
+
"acquisition_seed": null,
|
| 31 |
+
"candidate_n": 0,
|
| 32 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 33 |
+
"candidate_seed": null,
|
| 34 |
+
"energy_mae": 0.006517067205983039,
|
| 35 |
+
"energy_rmse": 0.007017354304601934,
|
| 36 |
+
"evaluation_policy": "fixed_member",
|
| 37 |
+
"forces_mae": 0.009533511264421878,
|
| 38 |
+
"forces_rmse": 0.013129118546651844,
|
| 39 |
+
"is_acquisition_iteration": false,
|
| 40 |
"iteration": 9,
|
| 41 |
+
"labeled_n_after_acquisition": 550,
|
| 42 |
"max_disagreement": 0.0,
|
| 43 |
"mean_disagreement": 0.0,
|
| 44 |
"n_labeled": 550,
|
| 45 |
+
"selected_n": 0,
|
| 46 |
+
"train_n_before_acquisition": 550,
|
| 47 |
+
"training_policy": "scratch",
|
| 48 |
+
"training_time": 17430.808257341385
|
| 49 |
},
|
| 50 |
"packaged_files": [
|
| 51 |
{
|
|
|
|
| 64 |
"raw_file": "committee_2-best.ckpt"
|
| 65 |
}
|
| 66 |
],
|
| 67 |
+
"published_files": [
|
| 68 |
+
"committee_0-best.ckpt",
|
| 69 |
+
"committee_1-best.ckpt",
|
| 70 |
+
"committee_2-best.ckpt"
|
| 71 |
+
],
|
| 72 |
"raw_files": [
|
| 73 |
{
|
| 74 |
"published_file": "committee_0-best.ckpt",
|
| 75 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt"
|
| 76 |
},
|
| 77 |
{
|
| 78 |
"published_file": "committee_1-best.ckpt",
|
| 79 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt"
|
| 80 |
},
|
| 81 |
{
|
| 82 |
"published_file": "committee_2-best.ckpt",
|
| 83 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
|
| 84 |
}
|
| 85 |
],
|
| 86 |
+
"selected_seed": 3,
|
| 87 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 88 |
"source_files": [
|
| 89 |
+
"results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt",
|
| 90 |
+
"results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt",
|
| 91 |
+
"results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
|
| 92 |
],
|
| 93 |
"strategy": "qbc"
|
| 94 |
}
|
nequip-random/metadata.json
CHANGED
|
@@ -12,15 +12,26 @@
|
|
| 12 |
],
|
| 13 |
"final_iteration": 9,
|
| 14 |
"final_metrics": {
|
| 15 |
-
"
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
"iteration": 9,
|
|
|
|
| 20 |
"max_disagreement": 0.0,
|
| 21 |
"mean_disagreement": 0.0,
|
| 22 |
"n_labeled": 550,
|
| 23 |
-
"
|
|
|
|
|
|
|
|
|
|
| 24 |
},
|
| 25 |
"packaged_files": [
|
| 26 |
{
|
|
@@ -29,16 +40,19 @@
|
|
| 29 |
"raw_file": "nequip_single_0-best.ckpt"
|
| 30 |
}
|
| 31 |
],
|
|
|
|
|
|
|
|
|
|
| 32 |
"raw_files": [
|
| 33 |
{
|
| 34 |
"published_file": "nequip_single_0-best.ckpt",
|
| 35 |
-
"source_file": "results/
|
| 36 |
}
|
| 37 |
],
|
| 38 |
-
"selected_seed":
|
| 39 |
-
"source_commit": "
|
| 40 |
"source_files": [
|
| 41 |
-
"results/
|
| 42 |
],
|
| 43 |
"strategy": "random"
|
| 44 |
}
|
|
|
|
| 12 |
],
|
| 13 |
"final_iteration": 9,
|
| 14 |
"final_metrics": {
|
| 15 |
+
"acquisition_rule": "random",
|
| 16 |
+
"acquisition_seed": null,
|
| 17 |
+
"candidate_n": 0,
|
| 18 |
+
"candidate_policy": "proposal_stream_filtered",
|
| 19 |
+
"candidate_seed": null,
|
| 20 |
+
"energy_mae": 0.004235895300860648,
|
| 21 |
+
"energy_rmse": 0.0050607236903155136,
|
| 22 |
+
"evaluation_policy": "fixed_member",
|
| 23 |
+
"forces_mae": 0.01065873935388155,
|
| 24 |
+
"forces_rmse": 0.017322629771456894,
|
| 25 |
+
"is_acquisition_iteration": false,
|
| 26 |
"iteration": 9,
|
| 27 |
+
"labeled_n_after_acquisition": 550,
|
| 28 |
"max_disagreement": 0.0,
|
| 29 |
"mean_disagreement": 0.0,
|
| 30 |
"n_labeled": 550,
|
| 31 |
+
"selected_n": 0,
|
| 32 |
+
"train_n_before_acquisition": 550,
|
| 33 |
+
"training_policy": "scratch",
|
| 34 |
+
"training_time": 4275.6661376953125
|
| 35 |
},
|
| 36 |
"packaged_files": [
|
| 37 |
{
|
|
|
|
| 40 |
"raw_file": "nequip_single_0-best.ckpt"
|
| 41 |
}
|
| 42 |
],
|
| 43 |
+
"published_files": [
|
| 44 |
+
"nequip_single_0-best.ckpt"
|
| 45 |
+
],
|
| 46 |
"raw_files": [
|
| 47 |
{
|
| 48 |
"published_file": "nequip_single_0-best.ckpt",
|
| 49 |
+
"source_file": "results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
|
| 50 |
}
|
| 51 |
],
|
| 52 |
+
"selected_seed": 2,
|
| 53 |
+
"source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
|
| 54 |
"source_files": [
|
| 55 |
+
"results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
|
| 56 |
],
|
| 57 |
"strategy": "random"
|
| 58 |
}
|