ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #3664 from hoba87/develop

Browse Source 
simplify execution of mliap pytorch example
This commit is contained in:
Axel Kohlmeyer
2023-03-28 18:05:22 -04:00
committed by GitHub
parent 7d5d2b909f 88760170bf
commit f9edd3a539
3 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/mliap/mliap_pytorch_Ta06A.py
View File

@ -94,8 +94,12 @@ lmp.commands_string(before_loading)
# Define the model however you like. In this example
# we load it from disk:
import os
import torch
model = torch.load('Ta06A.mliap.pytorch.model.pt')
torch_model = 'Ta06A.mliap.pytorch.model.pt'
if not os.path.exists(torch_model):
raise FileNotFoundError(f"Generate {torch_model} with convert_mliap_Ta06A.py")
model = torch.load(torch_model)
# Connect the PyTorch model to the mliap pair style.
lammps.mliap.load_model(model)

6
examples/mliap/mliap_pytorch_Ta06A_kokkos.py
View File

@ -94,8 +94,12 @@ lmp.commands_string(before_loading)
# Define the model however you like. In this example
# we load it from disk:
import os
import torch
model = torch.load('Ta06A.mliap.pytorch.model.pt')
torch_model = 'Ta06A.mliap.pytorch.model.pt'
if not os.path.exists(torch_model):
raise FileNotFoundError(f"Generate {torch_model} with convert_mliap_Ta06A.py")
model = torch.load(torch_model)
# Connect the PyTorch model to the mliap pair style.
lammps.mliap.load_model_kokkos(model)