search

atomistic-machine-learning / schnetpack

SchNetPack - Deep Neural Networks for Atomistic Systems
Other
791 stars 215 forks source link

Atomic energies #635

Closed rothjo closed 5 months ago

rothjo commented 5 months ago

Hi, is there a way to obtain atomic-wise energies from a SchNet?

Best, Jonas

jnsLs commented 5 months ago

Hi Jonas,

yes, you just need to specify a per_atom_output_key for the schnetpack.atomistic.Atomwise output module.

Best, Jonas

rothjo commented 5 months ago

Hi again,

ok perfect, thanks for the quick answer.

Best, Jonas