Skip to content

GitLab

Unverified Commit 13cacd04 authored by Ignacio Pickering's avatar Ignacio Pickering Committed by GitHub
Browse files

Make input more clear in example (#479)

parent af600d19
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/load_from_neurochem.py
View file @ 13cacd04
......@@ -68,8 +68,8 @@ coordinates = torch.tensor([[[0.03192167, 0.00638559, 0.01301679],
[0.45554739, 0.54289633, 0.81170881],
[0.66091919, -0.16799635, -0.91037834]]],
requires_grad=True)
species = consts.species_to_tensor('CHHHH').unsqueeze(0)
methane = ase.Atoms('CHHHH', positions=coordinates.squeeze().detach().numpy())
species = consts.species_to_tensor(['C', 'H', 'H', 'H', 'H']).unsqueeze(0)
methane = ase.Atoms(['C', 'H', 'H', 'H', 'H'], positions=coordinates.squeeze().detach().numpy())
###############################################################################
# Now let's compute energies using the ensemble directly:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment