Skip to content

Commit

Permalink
Readme instructions for local model
Browse files Browse the repository at this point in the history
  • Loading branch information
ijpulidos committed Sep 6, 2023
1 parent fde8a52 commit 89e3696
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,21 @@ espaloma_model(molecule_graph.heterograph)
openmm_system = esp.graphs.deploy.openmm_system_from_graph(molecule_graph)
```

If using espaloma from a local `.pt` file, say for example `espaloma-0.3.1.pt`,
then you would need to run the `eval` method of the model to get the correct
inference/predictions, as follows:

```python
import torch
...
# load local pretrained model
espaloma_model = torch.load("espaloma-0.3.1.pt")
espaloma_model.eval()
...
```

The rest of the code should be the same as in the previous code block example.

# Using espaloma to parameterize small molecules in relative free energy calculations

An example of using espaloma to parameterize small molecules in relative alchemical free energy calculations is provided in the `scripts/perses-benchmark/` directory.
Expand Down

0 comments on commit 89e3696

Please sign in to comment.