TorchSim Calculator Tutorial#
This tutorial demonstrates how to use a TACE model as a calculator within TorchSim.
TorchSim documentation: torchsim
For detailed usage and scripts (e.g., batchOPT, batchMD, and other scripts), see
xvzemin/tace
from tace.interface.torchsim import TACETorchSimCalc
# === Input ===
dtype = 'float32'
device = 'cuda' if torch.cuda.is_available() else 'cpu'
model = "../TACE-v1-OMat24-L.pt" # Your Model
fidelity_idx = 0 # first fidelity
model = TACETorchSimCalc(
model,
fidelity_idx=fidelity_idx,
device=device,
dtype=dtype,
compute_forces=True,
compute_stress=True,
)