Skip to content

Commit 196e061

Browse files
committed
benchmark file paths
1 parent 4f6b52b commit 196e061

File tree

3 files changed

+7
-8
lines changed

3 files changed

+7
-8
lines changed

validation/benchmarks/README.md

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ The code here has been used to produce the HDF5 benchmark result files (not incl
2020
- `Datasets/Industry/OpenFF_1.2.json` for small-molecule records.
2121
- `NMR_Data/J_coupling/...` and `NMR_Data/J_coupling/HBONDS/...` for NMR experimental references.
2222
- A simulation root for `nmr_analysis.py` with this layout:
23-
- `structures/gb3.pdb`, `structures/ubiquitin.pdb`, `structures/bpti.pdb`, `structures/hewl.pdb`
23+
- `structures/gb3.pdb`, `structures/ubiquitin.pdb`, `structures/bpti.pdb`, `structures/hewl.pdb`.
2424
- `dcd/<run_name>/<protein>_<replica>.dcd` (the script checks `dcd/64_7_ep12` first, then `dcd/espaloma`).
2525

2626
## Environment Setup
@@ -34,15 +34,15 @@ bash install_env.sh
3434
Notes:
3535

3636
- It is a GPU-oriented conda/pip setup and may need adaptation for your system. Some of the dependencies really do not like to play along together depending on the specific computer architecture.
37-
- `install_env.sh` currently uses `python Models/Espaloma/setup.py install`; that path to your local espaloma repo!
37+
- `install_env.sh` currently uses `python Models/Espaloma/setup.py install`, change that path to your local espaloma repo.
3838
- `src/run_model.py` contains a hardcoded Garnet checkpoint path (`trained_model_fp`); update it to your local model path before running `--backend garnet`.
3939

4040
## Small-Molecule Workflow
4141

4242
Run one command per backend:
4343

4444
```bash
45-
python main.py --backend <openff / espaloma / mace / garnet> --out <out file> --device <cpu / gpu / gpu:device> --n-workers <paraller workers>
45+
python main.py --backend <openff / espaloma / mace / garnet> --out <out file> --device <cpu / gpu / gpu:device> --n-workers <paraller workers>
4646
```
4747

4848
Optional flags:
@@ -54,7 +54,7 @@ The output HDF5 files store coordinates, energies, and geometry metrics (RMSD bo
5454

5555
## Small-Molecule Postprocessing Workflow
5656

57-
After the four backend HDF5 files are present. Make sure the names of the files generated in tge previous step match:
57+
After the four backend HDF5 files are present. Make sure the names of the files generated in the previous step match:
5858

5959
```bash
6060
python postprocess.py
@@ -108,4 +108,3 @@ This produces:
108108
- Many scripts use fixed filenames at the top of the file. If you change output names, update those constants.
109109
- The repo contains generated artifacts from previous runs.
110110
- Threading/device behavior is managed in `main.py`; adjust `THREADS_PER_SIM` and `--n-workers` to match your CPU/GPU memory budget.
111-

validation/benchmarks/nmr_analysis.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
parser.add_argument(
2424
"--simulations",
2525
type=str,
26-
default="/cephfs2/jgreener/dms/typing/val_folded/",
26+
default="./val_folded",
2727
help="Path to simulations master folder",
2828
)
2929
parser.add_argument(

validation/benchmarks/src/run_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from tempfile import NamedTemporaryFile
1515
from warnings import warn
1616

17-
trained_model_fp = Path("/lmb/home/jgreener/dms/typing/") / "params_txt/64_7_ep12/model.pt"
17+
trained_model_fp = "./params_txt/64_7_ep12/model.pt"
1818

1919
element_indices = {
2020
1 : 0 , 3 : 1 , 5 : 2 , 6 : 3 , 7 : 4 , 8 : 5 , # H Li B C N O
@@ -504,4 +504,4 @@ def topology_to_openmm_system(self, topology, nonbondedMethod=NoCutoff,
504504
)
505505
return system, top_openmm
506506

507-
garnet = Model(trained_model_fp)
507+
garnet = Model(trained_model_fp)

0 commit comments

Comments
 (0)